Home
last modified time | relevance | path

Searched refs:module_ (Results 1 – 25 of 163) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference_test.cc44 module_ = CreateNewVerifiedModule(); in DynamicDimensionInferenceTest()
50 DynamicDimensionInference::Run(module_.get(), handler)); in RunInference()
64 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAdd()
84 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAddTuple()
95 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetGe()
98 std::unique_ptr<HloModule> module_; member in xla::__anondf2baf250111::DynamicDimensionInferenceTest
112 module_->AddEntryComputation(builder.Build()); in TEST_F()
113 SCOPED_TRACE(module_->ToString()); in TEST_F()
116 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
133 module_->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dhlo_alias_analysis_test.cc44 module_ = CreateNewVerifiedModule(); in HloAliasAnalysisTest()
50 analysis_ = HloAliasAnalysis::Run(module_.get(), in RunAnalysis()
94 DependencyHloOrdering ordering(module_.get()); in AnyValuesInSameBufferInterfere()
111 std::unique_ptr<HloModule> module_; member in xla::__anona4abf5fc0111::HloAliasAnalysisTest
126 module_->AddEntryComputation(builder.Build()); in TEST_F()
127 SCOPED_TRACE(module_->ToString()); in TEST_F()
161 module_->AddEntryComputation(builder.Build()); in TEST_F()
162 SCOPED_TRACE(module_->ToString()); in TEST_F()
205 module_->AddEntryComputation(builder.Build()); in TEST_F()
206 SCOPED_TRACE(module_->ToString()); in TEST_F()
[all …]
Dhlo_live_range_test.cc39 HloLiveRangeTest() : module_(CreateNewVerifiedModule()) {} in HloLiveRangeTest()
43 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).ValueOrDie(); in Analyze()
45 module_->entry_computation()) in Analyze()
49 std::unique_ptr<HloModule> module_; member in xla::__anon60e75d0b0111::HloLiveRangeTest
95 module_->AddEntryComputation(builder.Build()); in TEST_F()
97 HloSchedule schedule(module_.get()); in TEST_F()
99 schedule.set_sequence(module_->entry_computation(), {paramA, paramX, mul}); in TEST_F()
124 module_->AddEntryComputation(builder.Build()); in TEST_F()
126 HloSchedule schedule(module_.get()); in TEST_F()
128 schedule.set_sequence(module_->entry_computation(), in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc48 HloDataflowAnalysisTest() : module_(CreateNewVerifiedModule()) {} in HloDataflowAnalysisTest()
55 EXPECT_TRUE(flatten.Run(module_.get()).ok()); in RunAnalysis()
57 HloDataflowAnalysis::Run(*module_, ssa_form, bitcast_defines_value) in RunAnalysis()
95 std::unique_ptr<HloModule> module_; member in xla::__anon587423c40111::HloDataflowAnalysisTest
113 module_->AddEntryComputation(builder.Build()); in TEST_P()
114 SCOPED_TRACE(module_->ToString()); in TEST_P()
162 module_->AddEntryComputation(builder.Build()); in TEST_P()
163 SCOPED_TRACE(module_->ToString()); in TEST_P()
218 module_->AddEntryComputation(builder.Build()); in TEST_P()
219 SCOPED_TRACE(module_->ToString()); in TEST_P()
[all …]
Dcopy_insertion_test.cc406 WhileCopyInsertionTest() : module_(CreateNewVerifiedModule()) {} in WhileCopyInsertionTest()
658 module_->AddEntryComputation(builder.Build()); in BuildWhileInstruction()
666 module_->AddEntryComputation(builder.Build()); in BuildWhileInstruction()
766 auto condition = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
768 auto body = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
774 module_->AddEntryComputation(builder->Build()); in BuildWhileInstructionWithCustomInit()
778 std::unique_ptr<HloModule> module_; member in xla::__anonf19bbe020111::WhileCopyInsertionTest
800 auto condition = module_->AddEmbeddedComputation( in TEST_F()
803 module_->AddEmbeddedComputation(BuildIndependentBodyComputation()); in TEST_F()
806 InsertCopies(module_.get()); in TEST_F()
[all …]
Ddynamic_padder_test.cc77 DynamicPadderTest() : HloTestBase() { module_ = CreateNewVerifiedModule(); } in DynamicPadderTest()
89 return padder.Run(module_.get()); in RunPadder()
106 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetScalarAddComputation()
109 std::unique_ptr<HloModule> module_; member in xla::__anon655019a10111::DynamicPadderTest
132 module_->AddEntryComputation(builder.Build()); in TEST_F()
135 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
162 module_ = GetHloModule(hlo_text); in TEST_F()
181 module_->entry_computation()->GetInstructionWithName("custom-call.1"); in TEST_F()
183 module_->entry_computation()->GetInstructionWithName("custom-call.2"); in TEST_F()
192 slice_to_dynamic = module_->entry_computation()->root_instruction(); in TEST_F()
[all …]
Dhlo_domain_verifier.cc32 : module_(module), verifier_(verifier) {} in RunContext()
41 HloModule* module_; member in xla::HloDomainVerifier::RunContext
49 for (HloComputation* computation : module_->computations()) { in PopulateDomainKinds()
68 for (HloComputation* computation : module_->computations()) { in Run()
/external/python/pybind11/tests/test_embed/
Dtest_interpreter.cpp54 auto module_ = py::module_::import("test_interpreter"); variable
55 REQUIRE(py::hasattr(module_, "DerivedWidget"));
57 auto locals = py::dict("hello"_a="Hello, World!", "x"_a=5, **module_.attr("__dict__"));
64 auto py_widget = module_.attr("DerivedWidget")("The question");
73 REQUIRE_NOTHROW(py::module_::import("widget_module"));
74 REQUIRE_THROWS_WITH(py::module_::import("throw_exception"),
76 REQUIRE_THROWS_WITH(py::module_::import("throw_error_already_set"),
110 REQUIRE(py::module_::import("widget_module").attr("add")(1, 2).cast<int>() == 3);
113 REQUIRE(py::module_::import("external_module").attr("A")(123).attr("value").cast<int>() == 123);
117 py::module_::import("external_module").attr("internals_at")().cast<uintptr_t>());
[all …]
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dir_loader.cpp35 module_(m), in IrLoader()
129 module_->AddFunction(std::move(function_)); in AddInstruction()
162 module_->AddCapability(std::move(spv_inst)); in AddInstruction()
164 module_->AddExtension(std::move(spv_inst)); in AddInstruction()
166 module_->AddExtInstImport(std::move(spv_inst)); in AddInstruction()
168 module_->SetMemoryModel(std::move(spv_inst)); in AddInstruction()
170 module_->AddEntryPoint(std::move(spv_inst)); in AddInstruction()
172 module_->AddExecutionMode(std::move(spv_inst)); in AddInstruction()
174 module_->AddDebug1Inst(std::move(spv_inst)); in AddInstruction()
176 module_->AddDebug2Inst(std::move(spv_inst)); in AddInstruction()
[all …]
Ddecoration_manager.cpp65 auto context = module_->context(); in RemoveDecorationsFrom()
140 decoration->Clone(module_->context())); in RemoveDecorationsFrom()
142 module_->AddAnnotationInst(std::move(new_inst)); in RemoveDecorationsFrom()
143 auto decoration_iter = --module_->annotation_end(); in RemoveDecorationsFrom()
351 if (!module_) return; in AnalyzeDecorations()
354 for (Instruction& inst : module_->annotations()) { in AnalyzeDecorations()
389 IRContext* ctx = module_->context(); in AddDecoration()
503 auto context = module_->context(); in CloneDecorations()
506 std::unique_ptr<Instruction> new_inst(inst->Clone(module_->context())); in CloneDecorations()
508 module_->AddAnnotationInst(std::move(new_inst)); in CloneDecorations()
[all …]
Dir_loader.h49 Module* module() const { return module_; } in module()
54 module_->SetHeader({magic, version, generator, bound, reserved}); in SetModuleHeader()
74 Module* module_; variable
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dir_loader.cpp35 module_(m), in IrLoader()
129 module_->AddFunction(std::move(function_)); in AddInstruction()
162 module_->AddCapability(std::move(spv_inst)); in AddInstruction()
164 module_->AddExtension(std::move(spv_inst)); in AddInstruction()
166 module_->AddExtInstImport(std::move(spv_inst)); in AddInstruction()
168 module_->SetMemoryModel(std::move(spv_inst)); in AddInstruction()
170 module_->AddEntryPoint(std::move(spv_inst)); in AddInstruction()
172 module_->AddExecutionMode(std::move(spv_inst)); in AddInstruction()
174 module_->AddDebug1Inst(std::move(spv_inst)); in AddInstruction()
176 module_->AddDebug2Inst(std::move(spv_inst)); in AddInstruction()
[all …]
Ddecoration_manager.cpp64 auto context = module_->context(); in RemoveDecorationsFrom()
140 decoration->Clone(module_->context())); in RemoveDecorationsFrom()
142 module_->AddAnnotationInst(std::move(new_inst)); in RemoveDecorationsFrom()
143 auto decoration_iter = --module_->annotation_end(); in RemoveDecorationsFrom()
348 if (!module_) return; in AnalyzeDecorations()
351 for (Instruction& inst : module_->annotations()) { in AnalyzeDecorations()
386 IRContext* ctx = module_->context(); in AddDecoration()
500 auto context = module_->context(); in CloneDecorations()
503 std::unique_ptr<Instruction> new_inst(inst->Clone(module_->context())); in CloneDecorations()
505 module_->AddAnnotationInst(std::move(new_inst)); in CloneDecorations()
[all …]
Dir_loader.h49 Module* module() const { return module_; } in module()
54 module_->SetHeader({magic, version, generator, bound, reserved}); in SetModuleHeader()
74 Module* module_; variable
/external/deqp-deps/SPIRV-Tools/source/opt/
Dir_loader.cpp35 module_(m), in IrLoader()
129 module_->AddFunction(std::move(function_)); in AddInstruction()
162 module_->AddCapability(std::move(spv_inst)); in AddInstruction()
164 module_->AddExtension(std::move(spv_inst)); in AddInstruction()
166 module_->AddExtInstImport(std::move(spv_inst)); in AddInstruction()
168 module_->SetMemoryModel(std::move(spv_inst)); in AddInstruction()
170 module_->AddEntryPoint(std::move(spv_inst)); in AddInstruction()
172 module_->AddExecutionMode(std::move(spv_inst)); in AddInstruction()
174 module_->AddDebug1Inst(std::move(spv_inst)); in AddInstruction()
176 module_->AddDebug2Inst(std::move(spv_inst)); in AddInstruction()
[all …]
Ddecoration_manager.cpp65 auto context = module_->context(); in RemoveDecorationsFrom()
140 decoration->Clone(module_->context())); in RemoveDecorationsFrom()
142 module_->AddAnnotationInst(std::move(new_inst)); in RemoveDecorationsFrom()
143 auto decoration_iter = --module_->annotation_end(); in RemoveDecorationsFrom()
351 if (!module_) return; in AnalyzeDecorations()
354 for (Instruction& inst : module_->annotations()) { in AnalyzeDecorations()
389 IRContext* ctx = module_->context(); in AddDecoration()
503 auto context = module_->context(); in CloneDecorations()
506 std::unique_ptr<Instruction> new_inst(inst->Clone(module_->context())); in CloneDecorations()
508 module_->AddAnnotationInst(std::move(new_inst)); in CloneDecorations()
[all …]
Dir_loader.h49 Module* module() const { return module_; } in module()
54 module_->SetHeader({magic, version, generator, bound, reserved}); in SetModuleHeader()
74 Module* module_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc32 : module_(CreateNewVerifiedModule()), in WhileTransformerTest()
105 module_->AddEntryComputation(builder.Build()); in BuildWhileInstruction()
119 std::unique_ptr<HloModule> module_; member in xla::__anon1ae185530111::WhileTransformerTest
128 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
129 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
139 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10)); in TEST_F()
140 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(1, 0, 1)); in TEST_F()
150 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5)); in TEST_F()
151 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
161 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
[all …]
/external/python/pybind11/include/pybind11/
Dembed.h49 static ::pybind11::module_::module_def \
51 static void PYBIND11_CONCAT(pybind11_init_, name)(::pybind11::module_ &); \
53 auto m = ::pybind11::module_::create_extension_module( \
65 void PYBIND11_CONCAT(pybind11_init_, name)(::pybind11::module_ &variable)
110 module_::import("sys").attr("path").cast<list>().append("."); in PYBIND11_NAMESPACE_END()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc46 module_(absl::make_unique<HloModule>("extracted", config_)), in ExtractionVisitor()
47 clone_context_(module_.get()), in ExtractionVisitor()
82 module_->AddEntryComputation(builder_.Build()); in FinishVisit()
97 HloModule* module() { return module_.get(); } in module()
99 std::unique_ptr<HloModule> ConsumeModule() { return std::move(module_); } in ConsumeModule()
104 std::unique_ptr<HloModule> module_; member in xla::__anonf1179c6c0111::ExtractionVisitor
/external/google-breakpad/src/processor/
Dminidump.cc1848 module_(), in MinidumpModule()
1877 if (!minidump_->ReadBytes(&module_, MD_MODULE_SIZE)) { in Read()
1883 Swap(&module_.base_of_image); in Read()
1884 Swap(&module_.size_of_image); in Read()
1885 Swap(&module_.checksum); in Read()
1886 Swap(&module_.time_date_stamp); in Read()
1887 Swap(&module_.module_name_rva); in Read()
1888 Swap(&module_.version_info.signature); in Read()
1889 Swap(&module_.version_info.struct_version); in Read()
1890 Swap(&module_.version_info.file_version_hi); in Read()
[all …]
/external/python/pybind11/tests/
Dpybind11_tests.cpp29 std::list<std::function<void(py::module_ &)>> &initializers() { in initializers()
30 static std::list<std::function<void(py::module_ &)>> inits; in initializers()
39 initializers().emplace_back([=](py::module_ &parent) { in test_initializer()
45 void bind_ConstructorStats(py::module_ &m) { in bind_ConstructorStats()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc39 hlo.parent() == module_.entry_computation()) { in AddAliasingInformationToIrArray()
54 if (module_.config().debug_options().xla_llvm_enable_alias_scope_metadata()) { in AddAliasingInformationToIrArray()
65 if (module_.config().debug_options().xla_llvm_enable_noalias_metadata()) { in AddAliasingInformationToIrArray()
76 if (module_.config() in AddAliasingInformationToIrArray()
82 hlo.parent() == module_.entry_computation()) { in AddAliasingInformationToIrArray()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_module_importer.cc36 module_(module), in HloModuleImporter()
47 *module.entry_computation(), module_, &function_map_, &builder_); in Import()
51 *computation, module_, &function_map_, &builder_)); in Import()
/external/webrtc/test/time_controller/
Dexternal_time_controller.cc89 : module_(module), thread_(thread) {} in ModuleWrapper()
92 return module_->TimeUntilNextProcess(); in TimeUntilNextProcess()
95 void Process() override { module_->Process(); } in Process()
99 module_->ProcessThreadAttached(thread_); in ProcessThreadAttached()
101 module_->ProcessThreadAttached(nullptr); in ProcessThreadAttached()
106 Module* module_; member in webrtc::ExternalTimeController::ProcessThreadWrapper::ModuleWrapper

1234567