Searched refs:upcast_integer (Results 1 – 1 of 1) sorted by relevance
1245 bool upcast_integer = false, in PropagateTensorShapeOnNode()1254 } else if (upcast_integer && !at::isFloatingType(*type->scalarType())) { in PropagateTensorShapeOnNode()1270 bool upcast_integer) -> type_vec_t { in PropagateTensorShapeOnNode() argument1275 node, *maybe_keepdim ? 0 : num_reduced_dim, upcast_integer); in PropagateTensorShapeOnNode()