Searched defs:num_compute_ops (Results 1 – 2 of 2) sorted by relevance
1629 const int64 num_compute_ops, const OpContext& op_context, in PredictDefaultNodeCosts()1669 int64 num_compute_ops = CountConv2DOperations(op_info, &found_unknown_shapes); in PredictConv2D() local1684 int64 num_compute_ops = CountConv2DBackpropInputOperations( in PredictConv2DBackpropInput() local1700 int64 num_compute_ops = CountConv2DBackpropFilterOperations( in PredictConv2DBackpropFilter() local1796 int64 num_compute_ops = CountMatMulOperations(op_info, &found_unknown_shapes); in PredictMatMul() local1928 int64 num_compute_ops = in PredictBatchMatMul() local
44 int64 num_compute_ops = 0; member