Home
last modified time | relevance | path

Searched refs:found_unknown_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc154 bool found_unknown_shapes; in CwiseOutputElementCount() local
157 MaybeGetMinimumShape(input_shape_1, rank, &found_unknown_shapes); in CwiseOutputElementCount()
161 MaybeGetMinimumShape(input_shape_1, rank, &found_unknown_shapes); in CwiseOutputElementCount()
163 MaybeGetMinimumShape(input_shape_2, rank, &found_unknown_shapes); in CwiseOutputElementCount()
184 int rank, bool* found_unknown_shapes) { in MaybeGetMinimumShape() argument
189 *found_unknown_shapes = true; in MaybeGetMinimumShape()
200 *found_unknown_shapes = true; in MaybeGetMinimumShape()
208 *found_unknown_shapes = true; in MaybeGetMinimumShape()
450 bool found_unknown_shapes = false; in PredictCwiseOp() local
454 int64 op_count = CalculateLargestInputCount(op_info, &found_unknown_shapes); in PredictCwiseOp()
[all …]
Dop_level_cost_estimator.h30 int rank, bool* found_unknown_shapes);
79 bool* found_unknown_shapes) const;
82 bool* found_unknown_shapes) const;
84 bool* found_unknown_shapes) const;
86 bool* found_unknown_shapes) const;
88 bool* found_unknown_shapes) const;
91 bool* found_unknown_shapes) const;
94 bool* found_unknown_shapes) const;
98 bool* found_unknown_shapes) const;
102 bool* found_unknown_shapes) const;
[all …]
Dop_level_cost_estimator_test.cc458 bool* found_unknown_shapes) const { in CountMatMulOperations()
459 return estimator_.CountMatMulOperations(op_info, found_unknown_shapes); in CountMatMulOperations()
463 bool* found_unknown_shapes) const { in CountBatchMatMulOperations()
464 return estimator_.CountBatchMatMulOperations(op_info, found_unknown_shapes); in CountBatchMatMulOperations()
491 bool found_unknown_shapes; in ValidateOpDimensionsFromImputs() local
494 &found_unknown_shapes); in ValidateOpDimensionsFromImputs()