| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 65 bool* slice_dim0, in SharedSliceValidation() argument
93 *slice_dim0 = true; in SharedSliceValidation()
115 (*slice_dim0) &= (i == 0) || take_all; in SharedSliceValidation()
128 bool slice_dim0 = true; in SharedSliceCommonCases() local
133 &slice_dim0, begin, size); in SharedSliceCommonCases()
142 if (slice_dim0 && in SharedSliceCommonCases()
|
| D | strided_slice_op.cc | 96 bool slice_dim0 = true; in Compute() local
108 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
122 if (slice_dim0 && IsDim0SliceAligned<T>(input.shape(), begin[0], end[0])) { in Compute()
204 bool slice_dim0 = true; in Compute() local
234 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
298 bool slice_dim0 = true; in Compute() local
344 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, 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.cc | 175 bool* is_identity, bool* is_simple_slice, bool* slice_dim0, in ValidateStridedSliceOp() argument
270 *slice_dim0 = true; in ValidateStridedSliceOp()
331 (*slice_dim0) &= (i == 0 && stride_i == 1) || take_all_in_dimension; in ValidateStridedSliceOp()
334 (*slice_dim0) &= (i == 0 && stride_i == 1) || begin_and_end_masked; in ValidateStridedSliceOp()
423 bool* slice_dim0, gtl::InlinedVector<int64, 4>* begin, in ValidateStridedSliceOp() argument
432 is_simple_slice, slice_dim0, begin, end, strides, shape_spec)); in ValidateStridedSliceOp()
|
| D | strided_slice_op.h | 70 bool* is_identity, bool* is_simple_slice, bool* slice_dim0,
82 bool* slice_dim0, gtl::InlinedVector<int64, 4>* begin,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1741 bool is_identity, is_simple_slice, slice_dim0; in __anon42d741192502() local
1747 &slice_dim0, &begin, &end, &strides)); in __anon42d741192502()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3370 bool slice_dim0; in GetStridedSliceAxis() local
3378 &slice_dim0, &slice_begin_vec, &slice_end_vec, &slice_strides_vec)); in GetStridedSliceAxis()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3145 bool slice_dim0; in ConvertStridedSlice() local
3153 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, &begin, &end, in ConvertStridedSlice()
|