Searched refs:CwiseOutputElementCount (Results 1 – 1 of 1) sorted by relevance
152 int64 CwiseOutputElementCount(const TensorShapeProto& input_shape_1, in CwiseOutputElementCount() function466 std::max(op_count, CwiseOutputElementCount(op_info.inputs(0).shape(), in PredictCwiseOp()