Searched refs:CreateModuleConfig (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | compile_only_client.cc | 27 CompileOnlyClient::CreateModuleConfig( in CreateModuleConfig() function in xla::CompileOnlyClient
31 return compiler_service_->CreateModuleConfig(program_shape, argument_shapes, in CreateModuleConfig()
|
| H A D | compile_only_client.h | 61 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_support.cc | 58 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig( in CreateModuleConfig() function
136 StatusOr<std::unique_ptr<xla::HloModuleConfig>> CreateModuleConfig( in CreateModuleConfig() function
142 return CreateModuleConfig(program_shape, argument_shapes, result_layout, in CreateModuleConfig()
324 CreateModuleConfig( in CreateHloModules()
|
| H A D | tpu_compile_op_support.h | 96 CreateModuleConfig(
108 CreateModuleConfig(
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | service.cc | 260 StatusOr<std::unique_ptr<HloModuleConfig>> Service::CreateModuleConfig( in CreateModuleConfig() function in xla::Service
272 return xla::CreateModuleConfig(program_shape, argument_shapes, in CreateModuleConfig()
277 StatusOr<std::unique_ptr<HloModuleConfig>> Service::CreateModuleConfig( in CreateModuleConfig() function in xla::Service
286 return CreateModuleConfig(program_shape, argument_shapes, &execution_options, in CreateModuleConfig()
661 CreateModuleConfig( in ExecuteGraphParallel()
876 CreateModuleConfig(ProgramShape{arg->computation().host_program_shape()}, in Compile()
|
| H A D | service.h | 187 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
196 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
|
| H A D | hlo_module_util.h | 36 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
|
| H A D | compile_only_service.cc | 122 CreateModuleConfig( in CompileAheadOfTime()
|
| H A D | hlo_module_util.cc | 44 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig( in CreateModuleConfig() function
|
| H A D | local_service.cc | 147 return CreateModuleConfig(program_shape, argument_layouts, in GetHloModuleConfig()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tfrt_cpu_pjrt_client.cc | 256 CreateModuleConfig(program_shape, argument_layouts, &execution_options, in JitCompile()
|