Home
last modified time | relevance | path

Searched defs:memrefs (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_kernels.cc237 llvm::SmallVectorImpl<MemrefDesc>* memrefs) { in ConvertTensorOperandsToMemrefDesc()
265 const llvm::SmallVectorImpl<MemrefDesc>& memrefs, in ExecuteImpl()
315 llvm::SmallVector<MemrefDesc> memrefs; in ExecuteImpl() local
Dtf_jitrt_kernels.cc606 std::vector<MemrefDesc> memrefs; in ConvertTensorOperandsToMemrefDesc() local
648 static void ExecuteImpl(Executable& executable, ArrayRef<MemrefDesc> memrefs, in ExecuteImpl()
708 auto memrefs = ConvertTensorOperandsToMemrefDesc(operands); in ExecuteImpl() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
Dtfrt_fallback.cc61 std::vector<MemrefDesc> memrefs; in RunTfrtFallback() local
Dtf_cpurt_executor.cc264 std::vector<MemrefDesc> memrefs(arguments.size()); in Execute() local
Dtf_jitrt_executor.cc208 std::vector<MemrefDesc> memrefs; in Execute() local
/external/tensorflow/tensorflow/compiler/xla/runtime/
Darguments_test.cc43 Arguments<MemrefDesc> memrefs(num_memrefs); in BM_CreateMemrefDesc_1d() local
Dsymbolic_shape_test.cc69 llvm::SmallVector<MemrefDesc> memrefs; in GetFakeMemrefs() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_memref_prints.cc176 auto memrefs = ExtractValuesToPrint(&op); in runOnOperation() local