Searched refs:ImportComputation (Results 1 – 2 of 2) sorted by relevance
166 TF_RETURN_IF_ERROR(ImportComputation(computation.proto(), &op.body())); in ReduceInternal()197 TF_RETURN_IF_ERROR(ImportComputation(computation.proto(), &op.body())); in ReduceWindowInternal()254 TF_RETURN_IF_ERROR(ImportComputation(comparator.proto(), &op.comparator())); in SortInternal()271 TF_RETURN_IF_ERROR(ImportComputation(condition.proto(), &op.cond())); in WhileInternal()272 TF_RETURN_IF_ERROR(ImportComputation(body.proto(), &op.body())); in WhileInternal()314 ImportComputation(update_computation.proto(), &op.update_computation())); in ScatterInternal()584 Status MlirHloBuilder::ImportComputation(const HloModuleProto& computation, in ImportComputation() function in xla::MlirHloBuilder
267 Status ImportComputation(const HloModuleProto& computation,