Home
last modified time | relevance | path

Searched defs:processing_shape (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstrided_slice_op.cc97 TensorShape processing_shape, final_shape; in Compute() local
205 TensorShape processing_shape, final_shape; in Compute() local
299 TensorShape processing_shape; // Without reshaping input. in Compute() local
H A Dstrided_slice_op_impl.h82 const TensorShape& processing_shape, in HandleStridedSliceCase()
120 const TensorShape& processing_shape, in HandleStridedSliceGradCase()
H A Dstrided_slice_op_test.cc106 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dstrided_slice_op.cc98 TensorShape processing_shape; in EmitDynamicSlice() local
430 PartialTensorShape processing_shape, final_shape; in CompileAsDynamicUpdateSlice() local
516 TensorShape processing_shape, final_shape; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dstrided_slice_op.cc175 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp()
427 TensorShape* processing_shape, TensorShape* final_shape, bool* is_identity, in ValidateStridedSliceOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc1773 PartialTensorShape processing_shape, final_shape; in __anon38bbb0e82502() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc2574 PartialTensorShape processing_shape; in ConvertSlice() local
2701 PartialTensorShape processing_shape; in ConvertStridedSlice() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc3950 TensorShape processing_shape; in GetStridedSliceAxis() local