| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 100 bool is_simple_slice = true; in Compute() local
111 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
152 if (is_simple_slice && std::is_same<Device, CPUDevice>::value && in Compute()
164 processing_shape, is_simple_slice, \ in Compute()
208 bool is_simple_slice = true; in Compute() local
237 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
264 is_simple_slice, result); \ in Compute()
303 bool is_simple_slice = true; in Compute() local
349 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, in Compute()
|
| D | strided_slice_op_impl.h | 45 bool is_simple_slice, Tensor* result);
53 bool is_simple_slice, Tensor* result);
83 bool is_simple_slice, Tensor* result) { in HandleStridedSliceCase() argument
87 if (is_simple_slice) { in HandleStridedSliceCase()
121 bool is_simple_slice, Tensor* result) { in HandleStridedSliceGradCase() argument
251 const TensorShape& processing_shape, bool is_simple_slice, \
257 const TensorShape& processing_shape, bool is_simple_slice, \
|
| 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,
82 bool* is_simple_slice, bool* slice_dim0,
|
| D | strided_slice_op.cc | 176 bool* is_identity, bool* is_simple_slice, bool* slice_dim0, in ValidateStridedSliceOp() argument
275 *is_simple_slice = true; in ValidateStridedSliceOp()
311 (*is_simple_slice) &= stride_i == 1; in ValidateStridedSliceOp()
428 bool* is_simple_slice, bool* slice_dim0, in ValidateStridedSliceOp() argument
438 is_simple_slice, slice_dim0, begin, end, strides, shape_spec)); in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2577 bool is_simple_slice; in ConvertSlice() local
2625 &is_identity, &is_simple_slice, &slice_dim0, &begin, &end, &strides, in ConvertSlice()
2636 << "\n is simple_slice: " << is_simple_slice in ConvertSlice()
2704 bool is_simple_slice; in ConvertStridedSlice() local
2715 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, &begin, &end, in ConvertStridedSlice()
2727 << "\n is simple_slice: " << is_simple_slice in ConvertStridedSlice()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3953 bool is_simple_slice; in GetStridedSliceAxis() local
3961 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetStridedSliceAxis()
3964 if (!is_simple_slice) return OkStatus(); in GetStridedSliceAxis()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1774 bool is_identity, is_simple_slice, slice_dim0; in __anon091a71b72502() local
1779 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in __anon091a71b72502()
|