| /external/tensorflow/tensorflow/compiler/xla/runtime/ |
| D | arguments_test.cc | 87 const MemrefDesc& memref) { in BenchmarkVerifyMemrefOperand()
96 auto memref = GetFakeMemref({1}); in BM_VerifyMemref_1d() local
101 auto memref = GetFakeMemref({1, 2}); in BM_VerifyMemref_2d() local
106 auto memref = GetFakeMemref({1, 2, 3}); in BM_VerifyMemref_3d() local
111 auto memref = GetFakeMemref({1, 2, 3, 4}); in BM_VerifyMemref_4d() local
116 auto memref = GetFakeMemref({1, 2, 3, 4, 5}); in BM_VerifyMemref_5d() local
|
| D | arguments.cc | 100 const MemrefDesc& memref) { in VerifyMemrefArgument()
170 if (auto* memref = dyn_cast<MemrefType>(&type)) in Verify() local
225 if (auto* memref = dyn_cast<MemrefType>(&type)) in VerifyMemrefArgument() local
227 if (auto* memref = dyn_cast<UnrankedMemrefType>(&type)) in VerifyMemrefArgument() local
268 if (auto* memref = dyn_cast<MemrefType>(&type)) in Verify() local
|
| D | jit_executable.cc | 72 if (auto* memref = dyn_cast<MemrefType>(type)) in HasStaticShapeOperands() local
195 const MemrefDesc& memref = cast<MemrefDesc>(arguments[i]); in CombineWithValueConstraineOperands() local
245 if (auto* memref = dyn_cast<MemrefType>(type)) { in GetExecutable() local
|
| D | symbolic_shape.cc | 65 if (auto* memref = dyn_cast<MemrefType>(type)) { in SymbolicShapesResolver() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/ |
| D | tf_cpurt_executor.cc | 166 MemrefDesc* memref) { in ConvertPyArrayMemrefDesc()
190 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, rank>* memref) { in Sizes()
195 static llvm::ArrayRef<int64_t> Strides(StridedMemRefType<T, rank>* memref) { in Strides()
200 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, 0>* memref) { in Sizes()
205 static llvm::ArrayRef<int64_t> Strides(StridedMemRefType<T, 0>* memref) { in Strides()
229 auto* memref = static_cast<StridedMemRefType<T, rank>*>(memref_ptr); in Convert() local
|
| D | tf_jitrt_executor.cc | 124 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, rank>* memref) { in Sizes()
129 static llvm::ArrayRef<int64_t> Strides(StridedMemRefType<T, rank>* memref) { in Strides()
134 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, 0>* memref) { in Sizes()
139 static llvm::ArrayRef<int64_t> Strides(StridedMemRefType<T, 0>* memref) { in Strides()
172 auto* memref = static_cast<StridedMemRefType<T, rank>*>(memref_ptr); in Convert() local
|
| D | tfrt_fallback.cc | 70 for (const auto& memref : memrefs) { in RunTfrtFallback() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
| D | lhlo_elemental_utils.cc | 43 Value createLoadOrUseCachedValue(Location loc, OpBuilder* b, Value memref, in createLoadOrUseCachedValue()
70 Value memref = cast<LmhloOp>(op).getResultBuffer(); in noLoaderUser() local
84 Value memref = op->getOperand(i); in noLoaderUser() local
247 Value memref) { in createMemRef1DReinterpretCast()
262 void createOffsetStore(OpBuilder& b, Location loc, Value res, Value memref, in createOffsetStore()
268 memref::LoadOp createOffsetLoad(OpBuilder& b, Location loc, Value memref, in createOffsetLoad()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| D | specialization.cc | 50 const MemrefDesc& memref) { in VerifyMemrefOperand()
86 if (auto memref = type.dyn_cast<mlir::MemRefType>()) { in SpecializeOperandType() local
226 auto* memref = dyn_cast<MemrefDesc>(&arguments[i]); in SpecializeFunction() local
|
| D | type_converter.cc | 63 if (auto memref = type.dyn_cast<mlir::MemRefType>()) { in ConvertCanonicalType() local
69 if (auto memref = type.dyn_cast<mlir::UnrankedMemRefType>()) { in ConvertCanonicalType() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt.h | 80 static bool IsStaticStorageDuration(StridedMemRefType<T, rank>* memref) { in IsStaticStorageDuration()
104 auto* memref = static_cast<StridedMemRefType<T, rank>*>(memref_ptr); in Convert() local
|
| D | tf_jitrt.h | 99 inline bool IsStaticStorageDuration(StridedMemRefType<T, rank>* memref) { in IsStaticStorageDuration()
122 auto* memref = static_cast<StridedMemRefType<T, rank>*>(memref_ptr); in Convert() local
|
| D | tf_jitrt_test.cc | 55 StridedMemRefType<float, 4> memref{ in BM_ReturnTensor() local
|
| D | tf_cpurt_passes.cc | 78 mlir::Value memref; member
320 mlir::Value memref, mlir::Value size, in MemrefToVector()
344 mlir::Value memref) { in MemrefToScalar()
|
| D | tf_cpurt_kernels.cc | 216 MemrefDesc* memref) { in ConvertTensorToMemrefDesc()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/utils/ |
| D | codegen_utils.cc | 30 Value emitNumElementsComputation(OpBuilder& b, Location loc, Value memref) { in emitNumElementsComputation()
85 Value linearIndex, Value memref) { in calcMultiDimIndex()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_memref_prints.cc | 82 void EmitMemRefPrint(Location loc, Value memref, OpBuilder* b) { in EmitMemRefPrint()
180 for (auto memref : memrefs) { in runOnOperation() local
|
| D | tensorflow_abi_knowledge_propagation.cc | 98 auto memref = operand.getType().dyn_cast<MemRefType>(); in runOnOperation() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| D | propagate_static_shapes_to_kernel.cc | 92 MemRefType memref, Type pointerType, in replaceStaticMemRefArguments()
167 auto memref = operands.getTypes().front().dyn_cast<MemRefType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | benchmark.cc | 48 auto* memref = static_cast<StridedMemRefType<float, 0>*>(result_ptr); in FreeReturnedMemref() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_buffer_forwarding.cc | 69 mlir::Value memref; member
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| D | bufferizable_op_interface_impl.cc | 237 FailureOr<Value> materializeExtraction(OpBuilder &b, Value memref, Value set) { in materializeExtraction()
265 Value memref, in materializeInsertion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | jitrt_custom_calls.cc | 236 static se::DeviceMemoryBase GetDeviceAddress(MemrefArg& memref) { in GetDeviceAddress()
242 static se::DeviceMemoryBase GetDeviceAddress(runtime::FlatMemrefView& memref) { in GetDeviceAddress()
307 static Shape ToShape(const runtime::StridedMemrefView& memref) { in ToShape()
452 auto memref = args.get<runtime::FlatMemrefView>(i); in operator ()() local
728 ArrayRef<int64_t> minor_to_major) { in GetConvDescriptor()
1529 if (auto memref = args.get<runtime::FlatMemrefView>(i); succeeded(memref)) { in operator ()() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/IR/ |
| D | lhlo_ops.cc | 197 Value memref = op.getOutput(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 120 void SetMemRef(mlir::Operation* op, mlir::Value memref) { in SetMemRef()
|