| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_reduce_sum_op_test.cc | 53 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
95 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
|
| D | sparse_dense_binary_op_shared_test.cc | 99 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
128 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
155 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
187 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
|
| D | sparse_add_op_test.cc | 63 const auto indices_shape = TensorShape({4, 2}); in TEST_F() local
|
| D | one_hot_op.cc | 55 const TensorShape& indices_shape = indices.shape(); in Compute() local
|
| D | ragged_gather_op_test.cc | 33 const TensorShape& indices_shape, const std::vector<INDEX_TYPE>& indices, in BuildRaggedGatherGraph()
|
| D | example_parsing_ops.cc | 682 TensorShape indices_shape({num_elements, 1}); in Compute() local
695 TensorShape indices_shape({0, 1}); in Compute() local
805 TensorShape indices_shape({total_num_features, 2}); in Compute() local
|
| D | gather_nd_op.cc | 96 const TensorShape& indices_shape(indices.shape()); in DoGatherNd() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 30 const xla::XlaOp& indices, const TensorShape& indices_shape, in XlaGather()
158 auto indices_shape = context->InputShape(1); in Compile() local
207 TensorShape indices_shape = context->InputShape(1); in Compile() local
|
| D | scatter_nd_op.cc | 33 const TensorShape& indices_shape, in ValidateUpdateShape()
84 TensorShape indices_shape = context->InputShape(0); in Compile() local
|
| D | one_hot_op.cc | 33 const TensorShape indices_shape = ctx->InputShape(0); in Compile() local
|
| D | sparse_to_dense_op.cc | 31 const TensorShape indices_shape = context->InputShape(0); in Compile() local
|
| D | segment_reduction_ops.cc | 57 TensorShape indices_shape = ctx->InputShape(1); in Compile() local
|
| D | dynamic_stitch_op.cc | 71 TensorShape indices_shape; in Compile() local
|
| D | tensor_array_ops.cc | 305 const TensorShape indices_shape = ctx->InputShape(1); in Compile() local
373 const TensorShape indices_shape = ctx->InputShape(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils_test.cc | 280 const Shape& indices_shape = args[1].shape(); in XLA_TEST_F() local
318 const Shape& indices_shape = args[1].shape(); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | resource_variable_ops.cc | 267 ShapeHandle indices_shape = c->input(1); in __anon3f9cfa090402() local
319 ShapeHandle indices_shape = c->input(1); in ResourceScatterUpdateShape() local
|
| D | state_ops.cc | 118 ShapeHandle indices_shape = c->input(1); in ScatterUpdateShape() local
|
| D | math_ops.cc | 1005 ShapeHandle indices_shape; in SparseSegmentReductionShapeFn() local
1029 ShapeHandle indices_shape; in SparseSegmentReductionGradShapeFn() local
1067 ShapeHandle indices_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
|
| D | array_ops.cc | 1101 ShapeHandle indices_shape = c->input(1); in __anon7c94107b1702() local
1121 ShapeHandle indices_shape = c->input(1); in __anon7c94107b1802() local
2874 Status ScatterNdShapeHelper(InferenceContext* c, ShapeHandle indices_shape, in ScatterNdShapeHelper()
2932 ShapeHandle indices_shape; in ScatterNdShape() local
2944 ShapeHandle indices_shape; in ScatterNdTensorShape() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 85 DataType index_type, const TensorShape& indices_shape, in OneHot()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense_test.cc | 31 SparseToDenseOpModel(std::initializer_list<int> indices_shape, in SparseToDenseOpModel()
|
| D | embedding_lookup_sparse_test.cc | 36 std::initializer_list<int> indices_shape, in EmbeddingLookupSparseOpModel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_op_test.py | 407 def testBatchDimsMatchesPythonBatching(self, params_shape, indices_shape, argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.h | 86 TensorShape indices_shape; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 646 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
671 Shape indices_shape = ShapeUtil::MakeShape(S32, {2}); in TEST_F() local
|