Searched defs:slice_begin (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | slice_op_test.py | 314 def _testGradientSlice(self, input_shape, slice_begin, slice_size): argument
339 def _testGradientSliceTape(self, input_shape, slice_begin, slice_size): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 253 absl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; in Compile() local
673 absl::InlinedVector<xla::XlaOp, 4> slice_begin; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1860 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedBoundRanges()
2028 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedShapeAndBoundRanges()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3231 NodeDef* slice_begin; in GetSimpleSliceAxis() local
3338 NodeDef* slice_begin; in GetStridedSliceAxis() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1808 const Tensor* slice_begin = ic->input_tensor(1); in MaybeUpdateNodeContextOutput() local
|