Home
last modified time | relevance | path

Searched refs:is_simple_slice (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op_impl.h46 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, \
Dstrided_slice_op.cc95 bool is_simple_slice = true; in Compute() local
106 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
147 if (is_simple_slice && std::is_same<Device, CPUDevice>::value && in Compute()
159 processing_shape, is_simple_slice, \ in Compute()
202 bool is_simple_slice = true; in Compute() local
231 &is_simple_slice, &slice_dim0, &begin, &end, &strides)); in Compute()
258 is_simple_slice, result); \ in Compute()
295 bool is_simple_slice = true; in Compute() local
327 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, in Compute()
352 is_simple_slice, old_lhs); \ in Compute()
Dstrided_slice_op_test.cc110 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/util/
Dstrided_slice_op.h49 bool* is_identity, bool* is_simple_slice, bool* slice_dim0,
59 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice,
Dstrided_slice_op.cc158 bool* is_identity, bool* is_simple_slice, bool* slice_dim0, in ValidateStridedSliceOp() argument
254 *is_simple_slice = true; in ValidateStridedSliceOp()
289 (*is_simple_slice) &= stride_i == 1; in ValidateStridedSliceOp()
380 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice, in ValidateStridedSliceOp() argument
389 is_simple_slice, slice_dim0, begin, end, strides)); in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3185 bool is_simple_slice; in GetSliceAxis() local
3193 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetSliceAxis()
3196 if (!is_simple_slice) return Status::OK(); in GetSliceAxis()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1602 bool is_identity, is_simple_slice, slice_dim0; in __anon7c94107b2502() local
1607 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in __anon7c94107b2502()