Home
last modified time | relevance | path

Searched refs:TypeIDNameRegistry (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/runtime/
Dtype_id.h32 class TypeIDNameRegistry {
34 using RegistrationFn = std::function<void(TypeIDNameRegistry&)>;
36 TypeIDNameRegistry() = default;
37 ~TypeIDNameRegistry() = default;
Dtype_id.cc22 llvm::StringRef TypeIDNameRegistry::FindTypeIDSymbolName(TypeID type_id) { in FindTypeIDSymbolName()
Dcustom_call.cc59 void PopulateCustomCallTypeIdNames(TypeIDNameRegistry& r) { in PopulateCustomCallTypeIdNames()
Dexecutable.cc70 DirectCustomCallLibrary lib, TypeIDNameRegistry::RegistrationFn types) { in ToSymbolsBinding()
75 TypeIDNameRegistry registry; in ToSymbolsBinding()
Dexecutable.h52 DirectCustomCallLibrary lib, TypeIDNameRegistry::RegistrationFn types = {});
Dcustom_call.h59 void PopulateCustomCallTypeIdNames(TypeIDNameRegistry& registry);
/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
Dpasses.h44 class TypeIDNameRegistry; variable
61 std::function<void(TypeIDNameRegistry&)> populate_type_id_names;
Dcompilation_pipeline.h32 std::function<void(TypeIDNameRegistry&)> populate_type_id_names;
Dcustom_call_encoding.h205 Globals(mlir::ModuleOp module, TypeIDNameRegistry type_id_names) in Globals()
265 TypeIDNameRegistry type_id_names_;
Drt_to_llvm.cc443 TypeIDNameRegistry type_id_names; in runOnOperation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Djitrt_custom_calls.h130 void PopulateXlaTypeIdNames(runtime::TypeIDNameRegistry& registry);
Djitrt_custom_calls.cc98 using ::xla::runtime::TypeIDNameRegistry;
116 void PopulateXlaTypeIdNames(TypeIDNameRegistry& registry) { in PopulateXlaTypeIdNames()