| /external/tensorflow/tensorflow/core/kernels/ |
| strided_slice_op_test.cc | 109 TensorShape processing_shape, final_shape; local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape,
|
| strided_slice_op.cc | 92 TensorShape processing_shape, final_shape; variable
105 &processing_shape, &final_shape, &is_identity,
138 const int processing_dims = processing_shape.dims();
140 if (processing_shape.num_elements() > 0) {
143 // TODO(aselle): Here we are restricting to processing_shape and
148 input_dims == 2 && processing_shape.dims() == 2 &&
159 processing_shape, is_simple_slice, \
199 TensorShape processing_shape, final_shape; variable
230 shrink_axis_mask, &processing_shape, &final_shape, &is_identity,
243 const int processing_dims = processing_shape.dims()
292 TensorShape processing_shape, final_shape; variable
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| strided_slice_op.cc | 129 TensorShape processing_shape, final_shape; variable
155 shrink_axis_mask_, &processing_shape, &final_shape, &dummy,
166 ctx, input_shape.dims() == processing_shape.dims(),
175 grad = xla::Reshape(grad, processing_shape.dim_sizes());
181 for (int i = 0; i < processing_shape.dims(); ++i) {
192 dims->edge_padding_low() + processing_shape.dim_size(i) +
193 (processing_shape.dim_size(i) - 1) * dims->interior_padding();
202 dims->edge_padding_high() + processing_shape.dim_size(i) +
203 (processing_shape.dim_size(i) - 1) * dims->interior_padding();
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| arithmetic_optimizer.cc | 3182 TensorShape processing_shape; local
[all...] |
