Searched refs:found_unknown_shapes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 323 int rank, bool* found_unknown_shapes) { in MaybeGetMinimumShape() argument
328 *found_unknown_shapes = true; in MaybeGetMinimumShape()
339 *found_unknown_shapes = true; in MaybeGetMinimumShape()
347 *found_unknown_shapes = true; in MaybeGetMinimumShape()
775 bool found_unknown_shapes = false; in PredictCwiseOp() local
779 int64 op_count = CalculateLargestInputCount(op_info, &found_unknown_shapes); in PredictCwiseOp()
785 CalculateTensorElementCount(op_info.outputs(0), &found_unknown_shapes)); in PredictCwiseOp()
801 &found_unknown_shapes, node_costs); in PredictCwiseOp()
808 bool found_unknown_shapes = false; in PredictCostOfAnUnknownOp() local
810 return PredictDefaultNodeCosts(0, op_context, &found_unknown_shapes, in PredictCostOfAnUnknownOp()
[all …]
|
| D | op_level_cost_estimator.h | 33 int rank, bool* found_unknown_shapes);
231 bool* found_unknown_shapes);
234 bool* found_unknown_shapes);
236 bool* found_unknown_shapes);
239 bool* found_unknown_shapes);
242 bool* found_unknown_shapes) const;
244 bool* found_unknown_shapes);
247 bool* found_unknown_shapes);
250 bool* found_unknown_shapes);
253 bool* found_unknown_shapes);
[all …]
|
| D | op_level_cost_estimator_test.cc | 520 bool* found_unknown_shapes) const { in CountMatMulOperations()
521 return estimator_.CountMatMulOperations(op_info, found_unknown_shapes); in CountMatMulOperations()
525 bool* found_unknown_shapes) const { in CountBatchMatMulOperations()
526 return estimator_.CountBatchMatMulOperations(op_info, found_unknown_shapes); in CountBatchMatMulOperations()
531 bool* found_unknown_shapes) const { in CountBatchMatMulOperations()
533 found_unknown_shapes); in CountBatchMatMulOperations()
560 bool found_unknown_shapes; in ValidateOpDimensionsFromInputs() local
563 &found_unknown_shapes); in ValidateOpDimensionsFromInputs()
604 bool* found_unknown_shapes) const { in CountBatchMatMulOperations()
606 op_info, found_unknown_shapes); in CountBatchMatMulOperations()
[all …]
|