HomeSort by relevance Sort by last modified time
    Searched refs:value_tensor (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
bias_op.cc 44 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format,
52 int32 channel_dim = value_tensor.dims() - 1;
53 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim));
55 *batch *= static_cast<int32>(value_tensor.dim_size(i));
58 int32 channel_dim = value_tensor.dims() - 3;
59 int32 height_dim = value_tensor.dims() - 2;
60 int32 width_dim = value_tensor.dims() - 1;
61 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim));
62 *height = static_cast<int32>(value_tensor.dim_size(height_dim));
63 *width = static_cast<int32>(value_tensor.dim_size(width_dim))
    [all...]
  /external/tensorflow/tensorflow/core/kernels/data/
random_dataset_op.cc 102 Tensor value_tensor(ctx->allocator({}), DT_INT64, {});
103 value_tensor.scalar<int64>()() = Random();
104 out_tensors->emplace_back(std::move(value_tensor));
range_dataset_op.cc 103 Tensor value_tensor(ctx->allocator({}), DT_INT64, {});
104 value_tensor.scalar<int64>()() = next_;
105 out_tensors->emplace_back(std::move(value_tensor));
  /external/tensorflow/tensorflow/python/ops/
resource_variable_ops.py 145 value_tensor = ops.convert_to_tensor(value)
146 shape.assert_is_compatible_with(value_tensor.shape)
148 value_tensor,
    [all...]
metrics_impl.py     [all...]
  /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
head.py 360 value_tensor, update_op = _identity_metric_single(
363 value_tensors.append(value_tensor)
  /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
head_test.py 69 value_tensor, update_tensor = model_fn_ops.eval_metric_ops[key]
76 value = value_tensor.eval()
    [all...]

Completed in 249 milliseconds