Searched refs:cast_dtype (Results 1 – 3 of 3) sorted by relevance
126 auto cast_dtype = AnfAlgo::GetOutputInferDataType(cast, 0); in ChangeNodeInferInfo() local134 (void)types.emplace_back(cast_dtype); in ChangeNodeInferInfo()143 (void)prim_op->AddAttr("cast_type", TypeIdToType(cast_dtype)); in ChangeNodeInferInfo()
308 TypePtr cast_dtype() { return cast_dtype_; } in cast_dtype() function
1181 const auto &cast_type = tensor->cast_dtype(); in DoParamMixPrecisionCast()