/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...] |