Home
last modified time | relevance | path

Searched refs:original_image_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.h288 const TensorShapeProto& original_image_shape,
294 const TensorShapeProto& original_image_shape, const OpInfo& op_info,
Dop_level_cost_estimator.cc891 const TensorShapeProto& original_image_shape, in ConvolutionDimensionsFromInputs() argument
895 VLOG(2) << "Original image shape: " << original_image_shape.DebugString(); in ConvolutionDimensionsFromInputs()
938 auto image_shape = MaybeGetMinimumShape(original_image_shape, in ConvolutionDimensionsFromInputs()
2124 const TensorShapeProto& original_image_shape, const OpInfo& op_info, in OpDimensionsFromInputs() argument
2127 VLOG(2) << "Original image shape: " << original_image_shape.DebugString(); in OpDimensionsFromInputs()
2129 MaybeGetMinimumShape(original_image_shape, 4, found_unknown_shapes); in OpDimensionsFromInputs()