| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_gpu.cc | 59 auto shaped_type = result.getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
75 auto shaped_type = input.getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
|
| D | legalize_to_linalg.cc | 1140 auto shaped_type = result_type.cast<ShapedType>(); in matchAndRewrite() local
1229 auto shaped_type = result_type.cast<ShapedType>(); in matchAndRewrite() local
1349 auto shaped_type = result_type.cast<ShapedType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | convert_type.cc | 149 auto shaped_type = type.dyn_cast<mlir::ShapedType>(); in GetShapeStrippedType() local
|
| D | constant_utils.cc | 28 PatternRewriter* rewriter, Location loc, ShapedType shaped_type, in CreateConstOpWithSingleValue()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | init_text_file_to_import_test_pass.cc | 69 ShapedType shaped_type = in runOnOperation() local
|
| D | readonly_references_to_resources.cc | 138 ShapedType shaped_type = in runOnFunction() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 351 mlir::RankedTensorType shaped_type, mlir::FloatType elem_type, in ConvertFloatBuffer()
413 mlir::RankedTensorType shaped_type, mlir::Type elem_type, in ConvertIntBuffer()
462 auto shaped_type = type.dyn_cast<mlir::RankedTensorType>(); in BuildExternalConstOp() local
501 mlir::RankedTensorType shaped_type, in BuildVariableOp()
528 auto shaped_type = type.dyn_cast<mlir::RankedTensorType>(); in BuildConstOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 219 if (auto shaped_type = elt_type.dyn_cast<mlir::ShapedType>()) { in ConvertAttribute() local
414 Status SetShapeAttribute(absl::string_view name, mlir::ShapedType shaped_type, in SetShapeAttribute()
|
| D | shape_inference_utils.cc | 354 if (auto shaped_type = in InferReturnTypeComponentsForTFOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_attributes.cc | 81 ShapeAttr ShapeAttr::get(mlir::MLIRContext* context, ShapedType shaped_type) { in get()
|
| D | tf_types.cc | 28 auto shaped_type = value.getType().cast<mlir::ShapedType>(); in GetShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 72 ShapedType shaped_type = in VerifyOperandsHaveSameShapesOrBroadcastableShape() local
326 if (auto shaped_type = t.dyn_cast_or_null<ShapedType>()) { in IsF32ShapedType() local
334 if (auto shaped_type = t.dyn_cast_or_null<ShapedType>()) { in IsBF16ShapedType() local
528 if (auto shaped_type = result_type.dyn_cast<RankedTensorType>()) { in buildComparisonBinOp() local
747 const auto shaped_type = value.getType().cast<ShapedType>(); in fold() local
2386 if (auto shaped_type = cond.getType().dyn_cast<ShapedType>()) { in BuildSelectV2Op() local
2392 if (auto shaped_type = operand_type.dyn_cast<ShapedType>()) { in BuildSelectV2Op() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 252 auto shaped_type = type.dyn_cast<ShapedType>(); in GetShapeStrippedType() local
861 if (auto shaped_type = t.dyn_cast<ShapedType>()) { in CanFuseAffineOp() local
|
| D | lower_static_tensor_list.cc | 347 if (auto shaped_type = element_shape.getType().dyn_cast<ShapedType>()) { in matchAndRewrite() local
369 if (auto shaped_type = in matchAndRewrite() local
|
| D | legalize_tf.cc | 105 auto shaped_type = value.getType().dyn_cast<ShapedType>(); in HasSameStaticShapes() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1120 auto shaped_type = type.cast<ShapedType>(); in inferReturnTypes() local
1733 auto shaped_type = input.getType().cast<ShapedType>(); in fold() local
2143 auto shaped_type = getType().cast<ShapedType>(); in fold() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 818 auto shaped_type = int_attr.getType(); in SliceDenseIntElementsAttrColumn2D() local
|