Searched refs:CreateModuleConfig (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | compile_only_client.cc | 26 CompileOnlyClient::CreateModuleConfig( in CreateModuleConfig() function in xla::CompileOnlyClient
30 return compiler_service_->CreateModuleConfig(program_shape, argument_shapes, in CreateModuleConfig()
|
| D | compile_only_client.h | 61 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.h | 95 CreateModuleConfig(
107 CreateModuleConfig(
|
| D | tpu_compile_op_support.cc | 56 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig( in CreateModuleConfig() function
134 StatusOr<std::unique_ptr<xla::HloModuleConfig>> CreateModuleConfig( in CreateModuleConfig() function
140 return CreateModuleConfig(program_shape, argument_shapes, result_layout, in CreateModuleConfig()
319 CreateModuleConfig( in CreateHloModules()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_util.h | 35 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
|
| D | service.h | 188 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
197 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
|
| D | service.cc | 255 StatusOr<std::unique_ptr<HloModuleConfig>> Service::CreateModuleConfig( in CreateModuleConfig() function in xla::Service
267 return xla::CreateModuleConfig(program_shape, argument_shapes, in CreateModuleConfig()
272 StatusOr<std::unique_ptr<HloModuleConfig>> Service::CreateModuleConfig( in CreateModuleConfig() function in xla::Service
281 return CreateModuleConfig(program_shape, argument_shapes, &execution_options, in CreateModuleConfig()
613 CreateModuleConfig( in ExecuteGraphParallel()
813 CreateModuleConfig(ProgramShape{arg->computation().host_program_shape()}, in Compile()
|
| D | compile_only_service.cc | 105 CreateModuleConfig( in CompileAheadOfTime()
|
| D | hlo_module_util.cc | 44 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig( in CreateModuleConfig() function
|
| D | local_service.cc | 151 CreateModuleConfig(program_shape, argument_layouts, &execution_options)); in CompileExecutables()
|