Home
last modified time | relevance | path

Searched defs:toTensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlegalize_to_tensor_op.cc49 auto toTensor = in matchAndRewrite() local
68 auto toTensor = shapeOf.getArg().getDefiningOp<bufferization::ToTensorOp>(); in matchAndRewrite() local
H A Dlhlo_fuse_linalg.cc103 if (auto toTensor = dyn_cast<bufferization::ToTensorOp>(definingOp)) { in runOnOperation() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Divalue_inl.h160 inline at::Tensor IValue::toTensor() && { in toTensor() function
177 inline at::Tensor& IValue::toTensor() & { in toTensor() function
183 inline const at::Tensor& IValue::toTensor() const& { in toTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
H A Dbufferize_tiled_loop.cc158 if (auto toTensor = toMemref.getTensor().getDefiningOp<ToTensorOp>()) { in findExistingSubview() local
293 auto toTensor = output.getDefiningOp<bufferization::ToTensorOp>(); in matchAndRewrite() local
/aosp_15_r20/external/executorch/extension/android/src/main/java/org/pytorch/executorch/
H A DEValue.java135 public Tensor toTensor() { in toTensor() method in EValue
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dgpu_fusion_rewrite.cc291 auto toTensor = in annotateLaunchFunc() local
/aosp_15_r20/external/pytorch/android/pytorch_android/src/main/java/org/pytorch/
H A DIValue.java252 public Tensor toTensor() { in toTensor() method in IValue