Searched refs:slice_dim0 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 158 bool* is_identity, bool* is_simple_slice, bool* slice_dim0, in ValidateStridedSliceOp() argument
253 *slice_dim0 = true; in ValidateStridedSliceOp()
314 (*slice_dim0) &= (i == 0 && stride_i == 1) || take_all_in_dimension; in ValidateStridedSliceOp()
317 (*slice_dim0) &= (i == 0 && stride_i == 1) || begin_and_end_masked; in ValidateStridedSliceOp()
381 bool* slice_dim0, gtl::InlinedVector<int64, 4>* begin, in ValidateStridedSliceOp() argument
389 is_simple_slice, slice_dim0, begin, end, strides)); in ValidateStridedSliceOp()
|
| D | strided_slice_op.h | 49 bool* is_identity, bool* is_simple_slice, bool* slice_dim0,
60 bool* slice_dim0, gtl::InlinedVector<int64, 4>* begin,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 67 bool* slice_dim0, in SharedValidation() argument
96 *slice_dim0 = true; in SharedValidation()
118 (*slice_dim0) &= (i == 0) || take_all; in SharedValidation()
131 bool slice_dim0 = true; in SharedSliceCommonCases() local
134 SharedValidation(context, output_shape, &is_identity, &slice_dim0, begin, in SharedSliceCommonCases()
145 if (slice_dim0 && in SharedSliceCommonCases()
|
| D | strided_slice_op.cc | 94 bool slice_dim0 = true; in Compute() local
106 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
120 if (slice_dim0 && IsDim0SliceAligned<T>(input.shape(), begin[0], end[0])) { in Compute()
201 bool slice_dim0 = true; in Compute() local
231 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
294 bool slice_dim0 = true; in Compute() local
327 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, in Compute()
|
| D | strided_slice_op_test.cc | 110 bool is_identity = true, slice_dim0 = true, is_simple_slice = true; in BM_ValidateStridedSliceOp() local
121 &is_identity, &is_simple_slice, &slice_dim0, &begin_out, &end_out, in BM_ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1602 bool is_identity, is_simple_slice, slice_dim0; in __anon7c94107b2502() local
1608 &slice_dim0, &begin, &end, &strides)); in __anon7c94107b2502()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3186 bool slice_dim0; in GetSliceAxis() local
3194 &slice_dim0, &slice_begin_vec, &slice_end_vec, &slice_strides_vec)); in GetSliceAxis()
|