Searched defs:orig_input_shape (Results 1 – 3 of 3) sorted by relevance
195 TensorShape orig_input_shape; in Compute() local
254 TensorShape orig_input_shape = orig_input_tensor.shape(); in Compute() local
2393 DimVector orig_input_shape(orig_input_shape_values.begin(), in matchAndRewrite() local