Searched defs:slice_begin (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | slice_op_test.py | 320 def _testGradientSlice(self, input_shape, slice_begin, slice_size): argument
345 def _testGradientSliceTape(self, input_shape, slice_begin, slice_size): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 290 absl::InlinedVector<int64_t, 4> slice_begin, slice_end, slice_strides; in Compile() local
712 absl::InlinedVector<xla::XlaOp, 4> slice_begin; in Compile() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | conv_spmd_expander.cc | 264 mlir::Value slice_begin = in HandleConv() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 161 SmallVector<int32_t, 4> slice_begin; in SliceOutputs() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2000 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedBoundRanges()
2169 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedShapeAndBoundRanges()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3815 NodeDef* slice_begin; in GetSimpleSliceAxis() local
3922 NodeDef* slice_begin; in GetStridedSliceAxis() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1847 const Tensor* slice_begin = ic->input_tensor(1); in MaybeUpdateNodeContextOutput() local
|