Searched refs:compilation_request (Results 1 – 5 of 5) sorted by relevance
342 TpuCompilationRequestProto compilation_request; in CreateTpuCompilationRequest() local344 compilation_request.set_use_mlir(use_mlir); in CreateTpuCompilationRequest()348 *compilation_request.mutable_mlir_module() = mlir_computation.mlir_module; in CreateTpuCompilationRequest()352 *compilation_request.mutable_fdef_lib() = in CreateTpuCompilationRequest()354 compilation_request.set_graph_def_version( in CreateTpuCompilationRequest()356 *compilation_request.mutable_function() = *function_computation.function; in CreateTpuCompilationRequest()364 *compilation_request.add_guaranteed_constants() = *constant; in CreateTpuCompilationRequest()372 compilation_request.add_guaranteed_constants()); in CreateTpuCompilationRequest()378 shape.AsProto(compilation_request.add_arg_shapes()); in CreateTpuCompilationRequest()381 *(compilation_request.mutable_metadata()) = metadata; in CreateTpuCompilationRequest()[all …]
34 TpuCompilationRequestProto compilation_request, in Compile()37 return TpuProgramGroup::CompileAndBuild(compilation_request, mesh_state, in Compile()
247 const TpuCompilationRequestProto& compilation_request, in CompileAndBuild() argument251 se_tpu::SerializeProto(compilation_request); in CompileAndBuild()268 count == compilation_request.metadata().num_cores_per_replica()); in CompileAndBuild()
94 const TpuCompilationRequestProto& compilation_request,
95 TpuSerializedProto compilation_request, const XLA_TpuMeshState* mesh_state,