Home
last modified time | relevance | path

Searched refs:value_tensor (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbias_op.cc46 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims() argument
55 int32 channel_dim = value_tensor.dims() - 1; in GetBiasValueDims()
56 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim)); in GetBiasValueDims()
58 *batch *= static_cast<int32>(value_tensor.dim_size(i)); in GetBiasValueDims()
61 *batch = static_cast<int32>(value_tensor.dim_size(0)); in GetBiasValueDims()
62 *channel = static_cast<int32>(value_tensor.dim_size(1)); in GetBiasValueDims()
63 *height = static_cast<int32>(value_tensor.dim_size(2)); in GetBiasValueDims()
64 if (value_tensor.dims() > 3) { in GetBiasValueDims()
65 *width = static_cast<int32>(value_tensor.dim_size(3)); in GetBiasValueDims()
67 if (value_tensor.dims() > 4) { in GetBiasValueDims()
[all …]
/external/tensorflow/tensorflow/python/data/util/
Dstructure_test.py172 value_tensor = constant_op.constant(42.0)
173 s_tensor = structure.Structure.from_value(value_tensor)
174 flat_tensor = s_tensor._to_tensor_list(value_tensor)
198 s_sparse_tensor._to_tensor_list(value_tensor)
206 s_nest._to_tensor_list(value_tensor)
/external/tensorflow/tensorflow/contrib/hadoop/kernels/
Dhadoop_dataset_ops.cc268 Tensor value_tensor(ctx->allocator({}), DT_STRING, {}); in GetNextInternal() local
269 value_tensor.scalar<string>()() = value; in GetNextInternal()
270 out_tensors->emplace_back(std::move(value_tensor)); in GetNextInternal()
/external/tensorflow/tensorflow/contrib/kinesis/kernels/
Dkinesis_dataset_ops.cc258 Tensor value_tensor(ctx->allocator({}), DT_STRING, {}); in GetNextInternal() local
259 value_tensor.scalar<std::string>()() = std::string(value); in GetNextInternal()
260 out_tensors->emplace_back(std::move(value_tensor)); in GetNextInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dlmdb_dataset_op.cc102 Tensor& value_tensor = out_tensors->back(); in GetNextInternal() local
103 value_tensor.scalar<string>()() = in GetNextInternal()
/external/tensorflow/tensorflow/python/ops/
Dresource_variable_ops.py271 value_tensor = ops.convert_to_tensor(value)
272 shape.assert_is_compatible_with(value_tensor.shape)
274 value_tensor,
1053 value_tensor = ops.convert_to_tensor(value, dtype=self.dtype)
1054 self._shape.assert_is_compatible_with(value_tensor.shape)
1056 self.handle, value_tensor, name=name)
Dmetrics_impl.py1484 value_tensor = _aggregate_variable(count, metrics_collections)
1490 return value_tensor, update_op
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_tpu_variables.py189 value_tensor = ops.convert_to_tensor(value, dtype=self.dtype)
191 self.handle, value_tensor, name=name)
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc238 Tensor value_tensor(input.dtype(), shape_1d); in GrapplerItemFromMetaGraphDef() local
239 InitializeTensor(input.dtype(), &value_tensor); in GrapplerItemFromMetaGraphDef()
241 NodeName(input.coo_sparse().values_tensor_name()), value_tensor); in GrapplerItemFromMetaGraphDef()
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dhead.py467 value_tensor, update_op = _identity_metric_single(
470 value_tensors.append(value_tensor)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dhead_test.py69 value_tensor, update_tensor = model_fn_ops.eval_metric_ops[key]
76 value = value_tensor.eval()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc3375 Tensor value_tensor(DT_FLOAT, TensorShape({})); in TEST_F() local
3376 value_tensor.flat<float>()(0) = 1; in TEST_F()
3377 value_tensor.AsProtoTensorContent(value.mutable_tensor()); in TEST_F()