Home
last modified time | relevance | path

Searched defs:tensor_ty (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.h88 if (auto tensor_ty = type.dyn_cast<TensorType>()) in IsValidTFTensorType() local
Dtf_ops.cc402 TensorType tensor_ty; in ParseTypeWithSubtype() local
Dtf_ops_n_z.cc580 auto tensor_ty = tensor.getType().cast<TensorType>(); in GetReshapeOutputType() local
679 auto tensor_ty = op.tensor().getType().cast<TensorType>(); in Verify() local
1929 auto tensor_ty = shape_op.input().getType().dyn_cast<RankedTensorType>(); in fold() local
2229 auto tensor_ty = op.tensor().getType().dyn_cast<RankedTensorType>(); in Verify() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc627 if (auto tensor_ty = type.dyn_cast<TFRTensorType>()) { in printType() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3923 auto tensor_ty = op.tensor().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
5502 auto tensor_ty = tensor.getType().cast<ShapedType>(); in matchAndRewrite() local