| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 92 TensorShape processing_shape, final_shape; in Compute() local
105 &processing_shape, &final_shape, &is_identity, in Compute()
138 const int processing_dims = processing_shape.dims(); in Compute()
140 if (processing_shape.num_elements() > 0) { in Compute()
148 input_dims == 2 && processing_shape.dims() == 2 && in Compute()
159 processing_shape, is_simple_slice, \ in Compute()
199 TensorShape processing_shape, final_shape; in Compute() local
230 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
243 const int processing_dims = processing_shape.dims(); in Compute()
247 if (processing_shape.dims() == 0) { in Compute()
[all …]
|
| D | strided_slice_op_impl.h | 45 const TensorShape& processing_shape,
53 const TensorShape& processing_shape,
62 const TensorShape& processing_shape, bool is_simple_slice,
83 const TensorShape& processing_shape, in HandleStridedSliceCase() argument
87 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in HandleStridedSliceCase()
121 const TensorShape& processing_shape, in HandleStridedSliceGradCase() argument
123 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in HandleStridedSliceGradCase()
145 const TensorShape& processing_shape, bool is_simple_slice, Tensor* result) { in operator()
146 gtl::InlinedVector<int64, 4> processing_dims = processing_shape.dim_sizes(); in operator()
169 const TensorShape& processing_shape, bool is_simple_slice, in operator()
[all …]
|
| D | strided_slice_op_test.cc | 109 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 129 TensorShape processing_shape, final_shape; in Compile() local
155 shrink_axis_mask_, &processing_shape, &final_shape, &dummy, in Compile()
166 ctx, input_shape.dims() == processing_shape.dims(), in Compile()
175 grad = xla::Reshape(grad, processing_shape.dim_sizes()); in Compile()
181 for (int i = 0; i < processing_shape.dims(); ++i) { in Compile()
192 dims->edge_padding_low() + processing_shape.dim_size(i) + in Compile()
193 (processing_shape.dim_size(i) - 1) * dims->interior_padding(); in Compile()
202 dims->edge_padding_high() + processing_shape.dim_size(i) + in Compile()
203 (processing_shape.dim_size(i) - 1) * dims->interior_padding(); in Compile()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 157 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp() argument
255 processing_shape->Clear(); in ValidateStridedSliceOp()
266 processing_shape->AddDim(shrink_i ? 1 : -1); in ValidateStridedSliceOp()
353 processing_shape->AddDim(size_i); in ValidateStridedSliceOp()
355 processing_shape->AddDim(-1); in ValidateStridedSliceOp()
367 final_shape->AddDim(processing_shape->dim_size(gather_index)); in ValidateStridedSliceOp()
379 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape, in ValidateStridedSliceOp() argument
392 if (!partial_processing_shape.AsTensorShape(processing_shape) || in ValidateStridedSliceOp()
|
| D | strided_slice_op.h | 48 PartialTensorShape* processing_shape, PartialTensorShape* final_shape,
58 int32 new_axis_mask, int32 shrink_axis_mask, TensorShape* processing_shape,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1601 PartialTensorShape processing_shape, final_shape; in __anon7c94107b2502() local
1607 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in __anon7c94107b2502()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3182 TensorShape processing_shape; in GetSliceAxis() local
3193 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetSliceAxis()
|