| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_config.cc | 31 HloModuleConfig::HloModuleConfig(const ProgramShape& program_shape, in HloModuleConfig() function in xla::HloModuleConfig
36 HloModuleConfig::HloModuleConfig(ComputationLayout entry_computation_layout) in HloModuleConfig() function in xla::HloModuleConfig
39 void HloModuleConfig::SetDefaultComputationLayout( in SetDefaultComputationLayout()
44 void HloModuleConfig::SetComputationLayoutIfExists( in SetComputationLayoutIfExists()
50 string HloModuleConfig::compilation_cache_key() const { in compilation_cache_key()
|
| D | elemental_ir_emitter_test.cc | 32 HloModuleConfig config; in RunTest()
40 HloModuleConfig config; in RunTypeConversionTest()
121 HloModuleConfig config; in XLA_TEST_F()
146 HloModuleConfig config; in XLA_TEST_F()
169 HloModuleConfig config; in XLA_TEST_F()
193 HloModuleConfig config; in XLA_TEST_F()
|
| D | hlo_module.h | 68 explicit HloModule(const string& name, HloModuleConfig config);
112 std::unique_ptr<HloModule> Clone(const HloModuleConfig& config,
229 const HloModuleConfig& config() const { return config_; } in config()
230 void set_config(const HloModuleConfig& config) { config_ = config; } in set_config()
242 const HloModuleProto& proto, const HloModuleConfig& module_config,
247 static StatusOr<HloModuleConfig> CreateModuleConfigFromProto(
253 static StatusOr<HloModuleConfig> CreateModuleConfigFromShape(
388 HloModuleConfig config_;
|
| D | hlo_runner_interface.cc | 25 HloModuleConfig config; in CreateModuleFromString()
35 TF_ASSIGN_OR_RETURN(HloModuleConfig config, in HloProtoToModule()
69 HloModuleConfig config; in ReadModuleFromHloTextFile()
82 HloModuleConfig module_config, in ReadModuleFromModuleBinaryProtofile()
|
| D | hlo_module_config.h | 41 class HloModuleConfig {
72 HloModuleConfig() = default;
74 explicit HloModuleConfig(const ProgramShape& program_shape,
77 explicit HloModuleConfig(ComputationLayout entry_computation_layout);
|
| D | dynamic_index_splitter_test.cc | 43 HloModuleConfig config; in TEST_F()
81 HloModuleConfig config; in TEST_F()
120 HloModuleConfig config; in TEST_F()
|
| D | service.h | 188 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
197 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
237 std::unique_ptr<HloModuleConfig> module_config, Backend* backend,
245 std::vector<std::unique_ptr<HloModuleConfig>> module_configs,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_unrolling_test.cc | 47 HloModuleConfig config; in TEST_F()
65 HloModuleConfig config; in TEST_F()
89 HloModuleConfig config; in TEST_F()
110 HloModuleConfig config; in TEST_F()
142 HloModuleConfig config; in TEST_F()
176 HloModuleConfig config; in TEST_F()
210 HloModuleConfig config; in TEST_F()
235 HloModuleConfig config; in TEST_F()
260 HloModuleConfig config; in TEST_F()
|
| D | gpu_index_test.cc | 68 HloModuleConfig config; in TEST_F()
93 HloModuleConfig config; in TEST_F()
124 HloModuleConfig config; in TEST_F()
156 HloModuleConfig config; in TEST_F()
185 HloModuleConfig config; in TEST_F()
|
| D | sorting_test.cc | 38 HloModuleConfig ConfigWithoutLayoutAssignment() { in ConfigWithoutLayoutAssignment()
39 HloModuleConfig config; in ConfigWithoutLayoutAssignment()
|
| D | gpu_input_fusible_slice_test.cc | 32 HloModuleConfig ConfigWithoutLayoutAssignment() { in ConfigWithoutLayoutAssignment()
33 HloModuleConfig config; in ConfigWithoutLayoutAssignment()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_options.h | 27 bool OptimizeForSizeRequested(const HloModuleConfig& config);
28 bool VectorizedReduceDisabled(const HloModuleConfig& config);
29 bool ForceEnableExperimentalLlvmIrGemm(const HloModuleConfig& config);
30 absl::optional<int64> LlvmIrGemvTilingFactor(const HloModuleConfig& config);
32 const HloModuleConfig& config);
|
| D | cpu_options.cc | 36 bool OptimizeForSizeRequested(const HloModuleConfig& config) { in OptimizeForSizeRequested()
42 bool VectorizedReduceDisabled(const HloModuleConfig& config) { in VectorizedReduceDisabled()
48 absl::optional<int64> LlvmIrGemvTilingFactor(const HloModuleConfig& config) { in LlvmIrGemvTilingFactor()
60 bool ForceEnableExperimentalLlvmIrGemm(const HloModuleConfig& config) { in ForceEnableExperimentalLlvmIrGemm()
74 const HloModuleConfig& config) { in LlvmIrGemmTileSize()
|
| D | tiled_dot_emitter.h | 34 const HloModuleConfig& module_config);
41 const HloModuleConfig& module_config);
50 llvm::IRBuilder<>* b, const HloModuleConfig& module_config);
|
| D | elemental_ir_emitter.h | 32 CpuElementalIrEmitter(const HloModuleConfig& module_config, in CpuElementalIrEmitter()
55 const HloModuleConfig& hlo_module_config_;
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_module_loader.cc | 41 HloModuleConfig* config) { in OverrideConfig()
68 const std::function<void(HloModuleConfig*)>& config_modifier_hook) { in LoadModuleFromData()
73 HloModuleConfig config; in LoadModuleFromData()
103 TF_ASSIGN_OR_RETURN(HloModuleConfig config, in LoadModuleFromData()
119 const std::function<void(HloModuleConfig*)>& config_modifier_hook) { in LoadModuleFromFile()
|
| D | hlo_module_loader.h | 55 const std::function<void(HloModuleConfig*)>& config_modifier_hook = {});
75 const std::function<void(HloModuleConfig*)>& config_modifier_hook = {});
|
| D | prepare_reference_module.cc | 36 const std::function<void(HloModuleConfig*)>& config_modifier_hook, in PrepareReferenceModule()
46 HloModuleConfig reference_config = test_module.config(); in PrepareReferenceModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator.h | 32 BufferComparator(const Shape& shape, const HloModuleConfig& config);
48 HloModuleConfig config_;
|
| D | nvptx_compiler.cc | 64 const HloModuleConfig& config) { in CandidateCudaRoots()
70 const HloModuleConfig& hlo_module_config) { in PrintCantFindCudaMessage()
84 string GetLibdeviceDir(const HloModuleConfig& hlo_module_config) { in GetLibdeviceDir()
203 bool MaybeLoadPtxFromFile(const HloModuleConfig module_config, in MaybeLoadPtxFromFile()
301 NVPTXCompiler::CompileTargetBinary(const HloModuleConfig& module_config, in CompileTargetBinary()
343 int cc_minor, const HloModuleConfig& hlo_module_config, bool relocatable) { in CompileGpuAsmOrGetCachedResult()
|
| D | ir_emitter_nested.h | 43 const HloModuleConfig& hlo_module_config,
67 IrEmitterNested(const HloModuleConfig& hlo_module_config,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| D | gpu_backend_lib.h | 44 const HloModuleConfig& hlo_module_config, const string& libdevice_dir_path,
54 const HloModuleConfig& hlo_module_config, const string& rocdl_dir_path);
|
| D | gpu_backend_lib.cc | 145 const HloModuleConfig& hlo_module_config, absl::string_view feature_str) { in GetTargetMachine()
339 const HloModuleConfig& hlo_module_config, in NVPTXTargetModuleLinker()
367 const HloModuleConfig& hlo_module_config) { in NVPTXGetTargetMachine()
375 llvm::Module*, GpuVersion, const HloModuleConfig&, const string&)>;
378 const HloModuleConfig& hlo_module_config, in LinkAndOptimizeModule()
464 void NVPTXBackendInit(const HloModuleConfig& hlo_module_config) { in NVPTXBackendInit()
514 const HloModuleConfig& hlo_module_config, const string& libdevice_dir_path, in CompileToPtx()
787 const HloModuleConfig& hlo_module_config, in AMDGPUTargetModuleLinker()
856 const HloModuleConfig& hlo_module_config) { in AMDGPUGetTargetMachine()
866 void AMDGPUBackendInit(const HloModuleConfig& hlo_module_config) { in AMDGPUBackendInit()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_program_group.h | 61 const std::vector<xla::HloModuleConfig::ShardableValueUpdatePair>
66 std::vector<xla::HloModuleConfig::ShardableValueUpdatePair> pairs) { in set_shardable_value_update_pairs()
83 std::vector<xla::HloModuleConfig::ShardableValueUpdatePair>
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_topk_test.cc | 39 HloModuleConfig config(program_shape); in TEST_F()
66 HloModuleConfig config(program_shape); in TEST_F()
|