Searched refs:partial_processing_shape (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | strided_slice_op.cc | 60 PartialTensorShape partial_processing_shape, in EmitDynamicSlice() argument
91 for (int64_t i = 0; i < partial_processing_shape.dims(); ++i) { in EmitDynamicSlice()
92 if (partial_processing_shape.dim_size(i) == -1) { in EmitDynamicSlice()
95 partial_processing_shape.set_dim(i, input_shape.dim_size(i)); in EmitDynamicSlice()
100 ctx, partial_processing_shape.AsTensorShape(&processing_shape), in EmitDynamicSlice()
103 partial_processing_shape.DebugString(), in EmitDynamicSlice()
262 PartialTensorShape partial_processing_shape, partial_final_shape; in Compile() local
271 shrink_axis_mask_, &partial_processing_shape, &partial_final_shape, in Compile()
387 EmitDynamicSlice(ctx, strides, partial_processing_shape, in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | strided_slice_op.cc | 433 PartialTensorShape partial_processing_shape, partial_final_shape; in ValidateStridedSliceOp() local
437 &partial_processing_shape, &partial_final_shape, is_identity, in ValidateStridedSliceOp()
441 if (!partial_processing_shape.AsTensorShape(processing_shape) || in ValidateStridedSliceOp()
444 partial_processing_shape.DebugString(), " and ", in ValidateStridedSliceOp()
|