Home
last modified time | relevance | path

Searched defs:orig_input_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_avgpooling_op.cc205 TensorShape orig_input_shape; in Compute() local
H A Dmkl_maxpooling_op.cc260 TensorShape orig_input_shape = orig_input_tensor.shape(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc2791 DimVector orig_input_shape(orig_input_shape_values.begin(), in matchAndRewrite() local