Searched defs:updates_shape (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 36 const TensorShape& updates_shape) { in ValidateUpdateShape()
87 TensorShape updates_shape = context->InputShape(1); in Compile() local
144 TensorShape updates_shape = context->InputShape(2); in CompileTensorScatter() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | state_ops.cc | 144 ShapeHandle updates_shape; in ScatterNdUpdateShape() local
|
| D | array_ops.cc | 3096 ShapeHandle updates_shape; in ScatterNdTensorShape() local
3142 ShapeHandle updates_shape; in __anon42d741194802() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad_test.cc | 225 TensorShape updates_shape({4}); in TEST_F() local
235 TensorShape updates_shape({2, 4, 4}); in TEST_F() local
245 TensorShape updates_shape({4}); in TEST_F() local
256 TensorShape updates_shape({2, 4, 4}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 481 Shape updates_shape; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 3446 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
3540 const Shape& updates_shape, const ProgramShape& to_apply_shape, in InferScatterShape()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_nd_op.cc | 823 const TensorShape& updates_shape(updates.shape()); in PrepareAndValidateInputs() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 1276 self, shape, begin, end, strides, updates_shape, *args): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2325 ShapeHandle updates_shape, in ScatterNdShapeHelper()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1140 const RuntimeShape& updates_shape, in ScatterNd()
|