Searched refs:out_ty (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 258 RankedTensorType out_ty = in matchAndRewrite() local
260 if (!out_ty || !out_ty.hasStaticShape()) return failure(); in matchAndRewrite()
281 ArrayRef<int64_t> item_shape = out_ty.getShape().drop_front(1); in matchAndRewrite()
282 auto item_ty = RankedTensorType::get(item_shape, out_ty.getElementType()); in matchAndRewrite()
293 SmallVector<Value, 8> values(out_ty.getDimSize(0)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1946 if (RankedTensorType out_ty = op.getType().dyn_cast<RankedTensorType>()) { in Verify() local
1947 if (out_ty.getRank() == 0) { in Verify()
2019 auto out_ty = op.getType().cast<TensorType>(); in Verify() local
2021 RankedTensorType::get(expected_shape, out_ty.getElementType()); in Verify()
2023 if (!AreCastCompatible({out_ty, expected_out_ty})) { in Verify()
2712 Type out_ty = in build() local
2714 build(builder, result, out_ty, input, reduction_indices, keep_dims); in build()
|
| D | tf_ops_n_z.cc | 1543 Type out_ty = in build() local
1545 build(builder, result, out_ty, input, reduction_indices, keep_dims); in build()
|