Home
last modified time | relevance | path

Searched defs:hlo_module (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc63 auto hlo_module = in TEST_F() local
97 auto hlo_module = in TEST_F() local
146 auto hlo_module = in TEST_F() local
186 auto hlo_module = in TEST_F() local
227 auto hlo_module = in TEST_F() local
260 auto hlo_module = in TEST_F() local
293 auto hlo_module = in TEST_F() local
334 auto hlo_module = in TEST_F() local
371 auto hlo_module = in TEST_F() local
435 auto hlo_module = in TEST_F() local
[all …]
Dgpu_unrolling_test.cc51 auto hlo_module = in TEST_F() local
71 auto hlo_module = in TEST_F() local
91 auto hlo_module = in TEST_F() local
123 auto hlo_module = in TEST_F() local
154 auto hlo_module = in TEST_F() local
188 auto hlo_module = in TEST_F() local
222 auto hlo_module = in TEST_F() local
247 auto hlo_module = in TEST_F() local
285 auto hlo_module = in TEST_F() local
Dgpu_input_fusible_slice_test.cc63 auto hlo_module = in TEST_F() local
105 auto hlo_module = in TEST_F() local
152 auto hlo_module = in TEST_F() local
Dgpu_ldg_test.cc56 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
86 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
113 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_ftz_test.cc45 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateBinaryOpModule() local
60 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateUnaryOpModule() local
Dgpu_dyn_shape_test.cc35 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_copy_test.cc48 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_codegen_test.cc50 std::unique_ptr<VerifiedHloModule> hlo_module, absl::string_view pattern) { in CompileAndOptionallyVerifyPtx()
Dgpu_noalias_test.cc50 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc45 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
61 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
84 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
105 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
124 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
141 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
161 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
186 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
217 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
233 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
[all …]
Dcpu_gpu_fusion_test.cc91 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D() local
201 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
249 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
284 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
308 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
332 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
347 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
362 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
377 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
392 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
[all …]
Dcodegen_test_base.cc21 std::unique_ptr<HloModule> hlo_module) { in CompileToExecutable()
33 std::unique_ptr<HloModule> hlo_module, in CompileToAotCompilationResult()
Dllvm_compiler_test.cc45 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
51 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
113 auto hlo_module = CreateNewVerifiedModule(); in TestCompilerHooks() local
135 std::unique_ptr<HloModule> hlo_module = CreateNewVerifiedModule(); in TestMultiModuleCompilation() local
Dllvm_irgen_test_base.cc50 std::unique_ptr<HloModule> hlo_module, const string& pattern, in CompileAndVerifyIr()
73 std::unique_ptr<HloModule> hlo_module, const AotCompilationOptions& options, in CompileAheadOfTimeAndVerifyIr()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc145 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
179 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
215 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
245 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
272 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
302 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
336 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
359 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
382 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
411 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
[all …]
Dmap_inliner_test.cc62 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
96 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
134 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
166 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dxla_debug_info_manager.cc23 const ModuleIdentifier& module_id, std::shared_ptr<HloModule> hlo_module, in RegisterModule()
41 const ModuleIdentifier& module_id, std::shared_ptr<HloModule> hlo_module, in UnregisterModule()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc81 Status InterpreterCompiler::RunHloOptimization(HloModule* hlo_module) { in RunHloOptimization()
97 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
105 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
Dexecutable.cc42 std::unique_ptr<HloModule> hlo_module, in InterpreterExecutable()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h41 const HloModule* hlo_module, const BufferAssignment* buffer_assignment, in IrEmitterContext()
59 const HloModule& hlo_module() const { return *hlo_module_; } in hlo_module() function
Dgpu_compiler.cc150 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule()
359 Status GpuCompiler::PrepareHloModuleForIrEmitting(HloModule* hlo_module) { in PrepareHloModuleForIrEmitting()
403 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
514 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* executor, in RunHloPassesAndBufferAssignement()
549 HloModule* hlo_module, llvm::LLVMContext* llvm_context, in CompileModuleToLlvmIrImpl()
984 HloModule* hlo_module, llvm::LLVMContext* llvm_context, in CompileModuleToLlvmIr()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.h43 explicit TpuExecutableInterface(std::shared_ptr<HloModule> hlo_module) in TpuExecutableInterface()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate.cc94 auto hlo_module = std::move(hlo_module_error.ValueOrDie()); in HloTextToMlirHloTranslateFunction() local
164 auto hlo_module = computation.proto(); in ConvertMlirHloToHloViaBuilder() local
197 HloModule* hlo_module = statusOrHloModule.ValueOrDie().get(); in MlirHloToHloTextTranslateFunctionImpl() local
Dhlo_to_mlir_hlo.cc32 Status ConvertHloToMlirHlo(mlir::ModuleOp module, xla::HloModule* hlo_module, in ConvertHloToMlirHlo()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc91 std::shared_ptr<HloModule> hlo_module) in TpuExecutable()
222 XLA_HloModule hlo_module; in RunHloPasses() local
250 XLA_HloModule hlo_module; in RunBackend() local

123