Home
last modified time | relevance | path

Searched defs:found_unknown_shapes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator.cc324 int rank, bool* found_unknown_shapes) { in MaybeGetMinimumShape()
787 bool found_unknown_shapes = false; in PredictCwiseOp() local
820 bool found_unknown_shapes = false; in PredictCostOfAnUnknownOp() local
895 const OpInfo& op_info, bool* found_unknown_shapes) { in CountConv2DOperations()
905 bool* found_unknown_shapes) { in ConvolutionDimensionsFromInputs()
1008 bool* found_unknown_shapes) { in CountConv2DOperations()
1046 const OpInfo& op_info, bool* found_unknown_shapes) { in CountMatMulOperations()
1053 bool* found_unknown_shapes) { in CountMatMulOperations()
1247 const OpInfo& op_info, bool* found_unknown_shapes) { in CountBatchMatMulOperations()
1253 bool* found_unknown_shapes) { in CountBatchMatMulOperations()
[all …]
H A Dop_level_cost_estimator_test.cc561 bool found_unknown_shapes; in ValidateOpDimensionsFromInputs() local
612 bool found_unknown_shapes; in CallOpDimensionsFromInputs() local