Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc195 TensorShape orig_input_shape; in Compute() local
Dmkl_maxpooling_op.cc254 TensorShape orig_input_shape = orig_input_tensor.shape(); in Compute() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2393 DimVector orig_input_shape(orig_input_shape_values.begin(), in matchAndRewrite() local