| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op_impl.h | 46 bool is_simple_slice, Tensor* result);
54 bool is_simple_slice, Tensor* result);
62 const TensorShape& processing_shape, bool is_simple_slice,
84 bool is_simple_slice, Tensor* result) { in HandleStridedSliceCase() argument
88 if (is_simple_slice) { in HandleStridedSliceCase()
122 bool is_simple_slice, Tensor* result) { in HandleStridedSliceGradCase() argument
145 const TensorShape& processing_shape, bool is_simple_slice, Tensor* result) { in operator()
169 const TensorShape& processing_shape, bool is_simple_slice, in operator()
248 const TensorShape& processing_shape, bool is_simple_slice, \
254 const TensorShape& processing_shape, bool is_simple_slice, \
|
| D | strided_slice_op.cc | 97 bool is_simple_slice = true; in Compute() local
108 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
149 if (is_simple_slice && std::is_same<Device, CPUDevice>::value && in Compute()
161 processing_shape, is_simple_slice, \ in Compute()
205 bool is_simple_slice = true; in Compute() local
234 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
261 is_simple_slice, result); \ in Compute()
299 bool is_simple_slice = true; in Compute() local
344 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, in Compute()
369 is_simple_slice, old_lhs); \ in Compute()
|
| D | strided_slice_op_test.cc | 107 bool is_identity = true, slice_dim0 = true, is_simple_slice = true; in BM_ValidateStridedSliceOp() local
118 &is_identity, &is_simple_slice, &slice_dim0, &begin_out, &end_out, in BM_ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.h | 70 bool* is_identity, bool* is_simple_slice, bool* slice_dim0,
81 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice,
|
| D | strided_slice_op.cc | 175 bool* is_identity, bool* is_simple_slice, bool* slice_dim0, in ValidateStridedSliceOp() argument
271 *is_simple_slice = true; in ValidateStridedSliceOp()
306 (*is_simple_slice) &= stride_i == 1; in ValidateStridedSliceOp()
422 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice, in ValidateStridedSliceOp() argument
432 is_simple_slice, slice_dim0, begin, end, strides, shape_spec)); in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3369 bool is_simple_slice; in GetStridedSliceAxis() local
3377 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetStridedSliceAxis()
3380 if (!is_simple_slice) return Status::OK(); in GetStridedSliceAxis()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1741 bool is_identity, is_simple_slice, slice_dim0; in __anondb9326b22502() local
1746 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in __anondb9326b22502()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3144 bool is_simple_slice; in ConvertStridedSlice() local
3153 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, &begin, &end, in ConvertStridedSlice()
|