Searched refs:original_image_shape (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | op_level_cost_estimator.h | 165 const TensorShapeProto& original_image_shape, 171 const TensorShapeProto& original_image_shape, const OpInfo& op_info,
|
D | op_level_cost_estimator.cc | 573 const TensorShapeProto& original_image_shape, in ConvolutionDimensionsFromInputs() argument 577 VLOG(2) << "Original image shape: " << original_image_shape.DebugString(); in ConvolutionDimensionsFromInputs() 580 MaybeGetMinimumShape(original_image_shape, 4, found_unknown_shapes); in ConvolutionDimensionsFromInputs() 1440 const TensorShapeProto& original_image_shape, const OpInfo& op_info, in OpDimensionsFromInputs() argument 1443 VLOG(2) << "Original image shape: " << original_image_shape.DebugString(); in OpDimensionsFromInputs() 1445 MaybeGetMinimumShape(original_image_shape, 4, found_unknown_shapes); in OpDimensionsFromInputs()
|