Home
last modified time | relevance | path

Searched defs:specialization (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/runtime/
Dexecutable.h108 llvm::Optional<size_t> specialization() const { return specialization_; } in specialization() function
238 llvm::Optional<size_t> specialization, in Executable()
Djit_executable.h53 Specialization specialization = Specialization::kAlways; member
Djit_executable.cc308 size_t specialization = entry.size - 1; in GetExecutable() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_jitrt.cc46 std::optional<size_t> specialization, in RecordCompileTime()
Dtf_jitrt_kernels.cc345 JitExecutable::UserData user_data) { in CompileImpl()
/external/rust/crates/itertools/benches/
Dfold_specialization.rs22 mod specialization { module
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
Dtf_cpurt_executor.cc71 Specialization specialization) { in Compile()
Dtf_jitrt_executor.cc79 Specialization specialization, bool vectorize, bool codegen_transpose, in Compile()
/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
Djit_compiler.cc174 llvm::Optional<size_t> specialization) { in Compile()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc500 void RemovePushedDownConstInputs(const FunctionSpecialization& specialization, in RemovePushedDownConstInputs()
535 const FunctionSpecialization& specialization, const NodeDef& func_node, in RemovePushedDownConstInputTypes()
561 void RemoveUnusedOutputsTypes(const FunctionSpecialization& specialization, in RemoveUnusedOutputsTypes()
609 const FunctionSpecialization& specialization, in UpdateSpecializedFunctionNode()
/external/rust/crates/vulkano/src/pipeline/
Dcompute_pipeline.rs142 let specialization = ash::vk::SpecializationInfo { in with_unchecked_pipeline_layout() localVariable
/external/deqp/external/vulkancts/modules/vulkan/pipeline/
DvktPipelineSpecConstantTests.cpp546 const Specialization specialization (m_specConstants, m_packData); in iterate() local
679 const Specialization specialization (m_specConstants, m_packData); in iterate() local
/external/rust/crates/smallvec/src/
Dlib.rs1679 mod specialization; module