Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_matmul_op.cc71 const double add_cost = Eigen::TensorOpCost::AddCost<Scalar>(); in GetCostPerUnit() local
75 (2 * num_eqs - 2) * add_cost); in GetCostPerUnit()
Dtridiagonal_solve_op.cc81 const double add_cost = Eigen::TensorOpCost::AddCost<Scalar>(); in GetCostPerUnit() local
89 (add_cost + mult_cost) * (2.5 * num_rhss + 1.5)); in GetCostPerUnit()
92 (add_cost + mult_cost) * (2 * num_rhss + 1)); in GetCostPerUnit()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc2542 const auto add_cost = EIGEN_COST(scalar_sum_op<float>); in PredictResizeBilinear() local
2586 add_cost + mul_cost + sub_cost_float + cast_to_float_cost; in PredictResizeBilinear()
2599 ops += (add_cost * 3 + sub_cost_float * 3 + mul_cost * 3) * output_elements; in PredictResizeBilinear()
2632 const auto add_cost = EIGEN_COST(scalar_sum_op<float>); in PredictCropAndResize() local
2650 ops += (mul_cost * 2 + sub_cost + add_cost) * crop_height * num_boxes; in PredictCropAndResize()
2652 ops += (mul_cost * 2 + sub_cost + add_cost) * crop_height * crop_width * in PredictCropAndResize()
2663 (cast_to_float_cost * 4 + add_cost * 3 + sub_cost * 3 + mul_cost * 3) * in PredictCropAndResize()