| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_gather_op.cc | 36 int64 value_size, Tensor* values_out) { in WriteValueSlices() argument
39 auto values = values_out->flat_outer_dims<VALUE_TYPE, 2>(); in WriteValueSlices()
234 Tensor* values_out = nullptr; in WriteValues() local
238 context->allocate_output(values_index, values_shape, &values_out)); in WriteValues()
244 values_out); in WriteValues()
257 int64 value_size, Tensor* values_out) const = 0;
269 int64 value_size, Tensor* values_out) const override { in CallWriteValueSlices()
271 value_size, values_out); in CallWriteValueSlices()
|
| D | sparse_cross_op.cc | 143 Tensor* indices_out, Tensor* values_out) in OutputUpdater() argument
146 values_out_(values_out) {} in OutputUpdater()
321 Tensor* values_out; in Compute() local
325 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out, in Compute()
329 output_start_indices, indices_out, values_out); in Compute()
507 Tensor** values_out, Tensor** shape_out, in CreateOutputTensors() argument
526 values_out)); in CreateOutputTensors()
|
| D | topk_op.cc | 78 Tensor* values_out = nullptr; in Compute() local
80 context->allocate_output(0, output_shape, &values_out)); in Compute()
88 auto values = values_out->flat_inner_dims<T>(); in Compute()
|
| /external/libbrillo/brillo/dbus/ |
| D | data_serialization_unittest.cc | 443 std::vector<int64_t> values_out; in TEST() local
444 EXPECT_TRUE(PopValueFromReader(&reader, &values_out)); in TEST()
446 EXPECT_EQ(values, values_out); in TEST()
529 VariantDictionary values_out; in TEST() local
530 EXPECT_TRUE(PopValueFromReader(&reader, &values_out)); in TEST()
532 EXPECT_EQ(values.size(), values_out.size()); in TEST()
533 EXPECT_EQ(values["key1"].Get<uint8_t>(), values_out["key1"].Get<uint8_t>()); in TEST()
534 EXPECT_EQ(values["key2"].Get<bool>(), values_out["key2"].Get<bool>()); in TEST()
535 EXPECT_EQ(values["key3"].Get<int16_t>(), values_out["key3"].Get<int16_t>()); in TEST()
536 EXPECT_EQ(values["key4"].Get<uint16_t>(), values_out["key4"].Get<uint16_t>()); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 171 values_out = grad_output_slices.values
175 if values_out.shape.is_fully_defined():
177 values_out.shape.as_list()[1:])
178 values = array_ops.zeros(values_shape, dtype=values_out.dtype,
186 values = array_ops.zeros(values_shape, dtype=values_out.dtype,
|
| D | sparse_ops.py | 607 indices_out, values_out, shape_out = gen_sparse_ops.sparse_cross(
619 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session_test.py | 631 indices_out, values_out, shape_out = sp_out
633 self.assertAllEqual(values_out, values)
641 indices_out, values_out, shape_out = s.run(sp)
643 self.assertAllEqual(values_out, values)
646 (indices_out, values_out, shape_out), = s.run([sp])
648 self.assertAllEqual(values_out, values)
656 indices_out, values_out, shape_out = s.run({'sp': sp})['sp']
658 self.assertAllEqual(values_out, values)
661 (indices_out, values_out, shape_out), = s.run({'sp': [sp]})['sp']
663 self.assertAllEqual(values_out, values)
[all …]
|
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| D | sparse_feature_cross_op.py | 97 indices_out, values_out, shape_out = (
110 indices_out, values_out, shape_out = (
122 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
|
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| D | sparse_feature_cross_kernel.cc | 143 Tensor* indices_out, Tensor* values_out) in OutputUpdater() argument
146 values_out_(values_out) {} in OutputUpdater()
364 Tensor* values_out; in Compute() local
368 CreateOutputTensors(columns, batch_size, context, &indices_out, &values_out, in Compute()
372 updater(output_start_indices, indices_out, values_out); in Compute()
550 Tensor** values_out, Tensor** shape_out, in CreateOutputTensors() argument
569 values_out)); in CreateOutputTensors()
|