Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
Dlegalize_to_tensor_op.cc49 auto toTensor = in matchAndRewrite() local
68 auto toTensor = shapeOf.getArg().getDefiningOp<bufferization::ToTensorOp>(); in matchAndRewrite() local
Dlhlo_fuse_linalg.cc103 if (auto toTensor = dyn_cast<bufferization::ToTensorOp>(definingOp)) { in runOnOperation() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
Dbufferize_tiled_loop.cc158 if (auto toTensor = toMemref.getTensor().getDefiningOp<ToTensorOp>()) { in findExistingSubview() local
293 auto toTensor = output.getDefiningOp<bufferization::ToTensorOp>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
Dgpu_fusion_rewrite.cc291 auto toTensor = in annotateLaunchFunc() local