Searched defs:updates_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 34 const TensorShape& updates_shape) { in ValidateUpdateShape()
85 TensorShape updates_shape = context->InputShape(1); in Compile() 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/ |
| D | hlo_evaluator_typed_visitor.h | 2059 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateScatterIndices()
2078 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateWindowIndices()
2107 const Shape& updates_shape, const Literal* scatter_indices) in UpdateScatterIndexToInputIndex()
2237 const Shape& updates_shape) { in UpdateWindowIndexToInputIndex()
2320 const Shape& updates_shape = updates.shape(); in HandleScatter() local
|
| D | shape_inference.cc | 3071 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()
3165 const Shape& updates_shape, const ProgramShape& to_apply_shape, in InferScatterShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2875 ShapeHandle updates_shape, in ScatterNdShapeHelper()
2934 ShapeHandle updates_shape; in ScatterNdShape() local
2946 ShapeHandle updates_shape; in ScatterNdTensorShape() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_nd_op.cc | 559 const TensorShape& updates_shape(updates.shape()); in PrepareAndValidateInputs() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1546 ShapeHandle updates_shape; in ScatterNdUpdateShape() local
|