Searched defs:updates_shape (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | scatter_nd_op.cc | 36 const TensorShape& updates_shape, in ValidateUpdateShape()
93 TensorShape updates_shape = context->InputShape(1); in Compile() local
152 TensorShape updates_shape = context->InputShape(2); in CompileTensorScatter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad_test.cc | 233 TensorShape updates_shape({4}); in TEST_F() local
243 TensorShape updates_shape({2, 4, 4}); in TEST_F() local
253 TensorShape updates_shape({4}); in TEST_F() local
264 TensorShape updates_shape({2, 4, 4}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | scatter_nd_util.cc | 24 const TensorShape& updates_shape) { in ValidateScatterNdUpdateShape()
|
| H A D | scatter_nd_op.cc | 782 const TensorShape& updates_shape(updates.shape()); in PrepareAndValidateInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | state_ops.cc | 144 ShapeHandle updates_shape; in ScatterNdUpdateShape() local
|
| H A D | array_ops.cc | 3137 ShapeHandle updates_shape; in ScatterNdTensorShape() local
3183 ShapeHandle updates_shape; in __anon38bbb0e84702() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.h | 484 Shape updates_shape; member
|
| H A D | ir_emitter_unnested.cc | 5244 const Shape& updates_shape = root->operand(2)->shape(); in EmitScatter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | array_ops_test.py | 1401 self, shape, begin, end, strides, updates_shape, *args): argument
1417 strides, updates_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference.cc | 3565 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
3698 const Shape& updates_shape = *updates_shapes[operand_i]; in InferScatterShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 661 const RuntimeShape& updates_shape, in ScatterNd()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 2416 ShapeHandle updates_shape, in ScatterNdShapeHelper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 1194 std::vector<int64_t> updates_shape(indices_first); in ChooseScatterArguments() local
|