Searched refs:overlay_lib (Results 1 – 8 of 8) sorted by relevance
219 if (!options.overlay_lib && overlay_lib_def_) { in Instantiate()221 options_copy.overlay_lib = overlay_lib_def_; in Instantiate()384 const FunctionLibraryDefinition* overlay_lib = nullptr; // Not owned. member565 options.overlay_lib = lib_def; in CreateKernel()634 options.overlay_lib = lib_def; in InstantiateSymbolicGradient()701 options.overlay_lib ? options.overlay_lib : base_lib_def_; in Instantiate()737 item->overlay_lib = options.overlay_lib; in Instantiate()740 if (options.overlay_lib) { in Instantiate()742 new FunctionLibraryRuntimeOverlay(this, options.overlay_lib); in Instantiate()865 lib_def = (*item)->overlay_lib; in CreateItem()
183 const FunctionLibraryDefinition& overlay_lib) in MultiDeviceFunctionData()188 overlay_lib_(overlay_lib) {} in MultiDeviceFunctionData()
411 std::unique_ptr<FunctionLibraryDefinition> overlay_lib( in TEST_F() local415 options.overlay_lib = overlay_lib.get(); in TEST_F()466 std::unique_ptr<FunctionLibraryDefinition> overlay_lib( in TEST_F() local470 options.overlay_lib = overlay_lib.get(); in TEST_F()487 options.overlay_lib = overlay_lib_override.get(); in TEST_F()
543 options.overlay_lib == nullptr ? lib_def_ : options.overlay_lib; in InstantiateMultiDevice()683 opts.overlay_lib = &data->overlay_lib_; in InstantiateMultiDevice()
137 inst_opts.overlay_lib = ctx->function_library().get(); in Instantiate()
545 const FunctionLibraryDefinition* overlay_lib = nullptr; member
910 if (options.overlay_lib) { in Canonicalize()912 "_overlay_lib", "=", reinterpret_cast<uintptr_t>(options.overlay_lib))); in Canonicalize()
173 inst_ops.overlay_lib = nullptr; in ConstructFunctionHandle()