Searched refs:GetMlirTypes (Results 1 – 2 of 2) sorted by relevance
129 TF_RETURN_IF_ERROR(GetMlirTypes(computation.parameter_instructions(), &args)); in ImportAsFunc()130 TF_RETURN_IF_ERROR(GetMlirTypes({computation.root_instruction()}, &rets)); in ImportAsFunc()163 TF_RETURN_IF_ERROR(GetMlirTypes(computation.parameter_instructions(), &args)); in ImportAsRegion()517 TF_RETURN_IF_ERROR(GetMlirTypes( in ImportInstructionImpl()531 TF_RETURN_IF_ERROR(GetMlirTypes( in ImportInstructionImpl()876 tensorflow::Status HloFunctionImporter::GetMlirTypes( in GetMlirTypes() function in xla::HloFunctionImporter
127 Status GetMlirTypes(const std::vector<xla::HloInstruction*>& instructions,