Searched refs:out_ty (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 293 RankedTensorType out_ty = in matchAndRewrite() local
295 if (!out_ty || !out_ty.hasStaticShape()) return failure(); in matchAndRewrite()
316 ArrayRef<int64_t> item_shape = out_ty.getShape().drop_front(1); in matchAndRewrite()
317 auto item_ty = RankedTensorType::get(item_shape, out_ty.getElementType()); in matchAndRewrite()
328 SmallVector<Value, 8> values(out_ty.getDimSize(0)); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 2199 if (RankedTensorType out_ty = op.getType().dyn_cast<RankedTensorType>()) { in verify() local
2200 if (out_ty.getRank() == 0) { in verify()
2272 auto out_ty = op.getType().cast<TensorType>(); in verify() local
2274 RankedTensorType::get(expected_shape, out_ty.getElementType()); in verify()
2276 if (!AreCastCompatible({out_ty, expected_out_ty})) { in verify()
3115 Type out_ty = InferReductionOpType(input, reduction_indices, keep_dims); in build() local
3116 build(builder, result, out_ty, input, reduction_indices, keep_dims); in build()
|
| D | tf_ops_n_z.cc | 1682 Type out_ty = InferReductionOpType(input, reduction_indices, keep_dims); in build() local
1683 build(builder, result, out_ty, input, reduction_indices, keep_dims); in build()
|