| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_import.cc | 379 mlir::RankedTensorType shaped_type, mlir::FloatType elem_type, in ConvertFloatBuffer()
443 mlir::RankedTensorType shaped_type, mlir::Type elem_type, in ConvertIntBuffer()
496 auto shaped_type = type.dyn_cast<mlir::RankedTensorType>(); in BuildExternalConstOp() local
535 mlir::RankedTensorType shaped_type, in BuildVariableOp()
580 const mlir::RankedTensorType shaped_type, OpBuilder& builder, in BuildSparseConstOp()
655 auto shaped_type = type.dyn_cast<mlir::RankedTensorType>(); in BuildConstOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| H A D | utils.cc | 44 auto shaped_type = value.getType().dyn_cast<ShapedType>(); in HasStaticShape() local
51 auto shaped_type = value.getType().dyn_cast<ShapedType>(); in HasStaticShapeAtDims() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | group_assignment.cc | 84 auto shaped_type = mlir::RankedTensorType::get( in ToMLIR() local
110 mlir::ShapedType shaped_type = group_assignment_attr.getType(); in FromMLIR() local
|
| H A D | collectives.cc | 312 auto shaped_type = mlir::RankedTensorType::get( in EmitAllReduce() local
420 auto shaped_type = mlir::RankedTensorType::get( in CreateConstSrcTargetPair() local
|
| H A D | group_assignment_test.cc | 49 auto shaped_type = mlir::RankedTensorType::get( in CreateGroupAssignmentAttr() local
|
| H A D | shape_utils.cc | 143 if (auto shaped_type = in InferShapeOfTFOpWithCustomOperandConstantFn() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_legalize_i1_type.cc | 97 ShapedType shaped_type = int_attr.getType(); in isLegalAttribute() local
110 ShapedType shaped_type = int_attr.getType(); in convertAttribute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | legalize_tf.cc | 100 auto shaped_type = value.getType().dyn_cast<ShapedType>(); in HasSameStaticShapes() local
119 if (auto shaped_type = val.getType().dyn_cast<RankedTensorType>()) { in CreateCastToInt32() local
132 auto shaped_type = input.getType().cast<ShapedType>(); in GetShape() local
|
| H A D | optimize.cc | 316 auto shaped_type = type.dyn_cast<ShapedType>(); in GetShapeStrippedType() local
1257 if (auto shaped_type = t.dyn_cast<ShapedType>()) { in CanFuseAffineOp() local
|
| H A D | lower_static_tensor_list.cc | 481 if (auto shaped_type = element_shape.getType().dyn_cast<ShapedType>()) { in matchAndRewrite() local
503 if (auto shaped_type = in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | export_utils.cc | 230 if (auto shaped_type = elt_type.dyn_cast<mlir::ShapedType>()) { in ConvertAttribute() local
443 Status SetShapeAttribute(absl::string_view name, mlir::ShapedType shaped_type, in SetShapeAttribute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | convert_attributes.cc | 181 if (auto shaped_type = elt_type.dyn_cast<ShapedType>()) { in ConvertAttribute() local
280 Status SetShapeAttribute(absl::string_view name, ShapedType shaped_type, in SetShapeAttribute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| H A D | tfl_ops.cc | 165 ShapedType shaped_type = in VerifyOperandsHaveSameShapesOrBroadcastableShape() local
468 if (auto shaped_type = t.dyn_cast_or_null<ShapedType>()) { in IsF32ShapedType() local
476 if (auto shaped_type = t.dyn_cast_or_null<ShapedType>()) { in IsBF16ShapedType() local
670 if (auto shaped_type = result_type.dyn_cast<RankedTensorType>()) { in buildComparisonBinOp() local
899 const auto shaped_type = value.getType().cast<ShapedType>(); in fold() local
3007 if (auto shaped_type = cond.getType().dyn_cast<ShapedType>()) { in BuildSelectV2Op() local
3013 if (auto shaped_type = operand_type.dyn_cast<ShapedType>()) { in BuildSelectV2Op() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | ensure_static_shapes_pass.cc | 32 if (ShapedType shaped_type = type.dyn_cast<ShapedType>()) { in runOnFunction() local
|
| H A D | freeze_saved_model_assets.cc | 92 ShapedType shaped_type = in runOnOperation() local
|
| H A D | init_text_file_to_import_test_pass.cc | 82 ShapedType shaped_type = in runOnOperation() local
|
| H A D | readonly_references_to_resources.cc | 150 ShapedType shaped_type = in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/common/ |
| H A D | utils.h | 49 if (auto shaped_type = t.dyn_cast_or_null<ShapedType>()) { in IsF32ShapedType() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| H A D | convert_type.cc | 194 auto shaped_type = type.dyn_cast<mlir::ShapedType>(); in GetShapeStrippedType() local
|
| H A D | constant_utils.cc | 32 PatternRewriter* rewriter, Location loc, ShapedType shaped_type, in CreateConstOpWithSingleValue()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/types/ |
| H A D | dialect.cc | 441 ShapeAttr ShapeAttr::get(MLIRContext *context, ShapedType shaped_type) { in get()
474 auto shaped_type = value.getType().cast<ShapedType>(); in GetShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | shape_simplification.cc | 197 auto shaped_type = in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| H A D | collective_lowering.cc | 597 const mlir::RankedTensorType shaped_type = mlir::RankedTensorType::get( in LowerAllGatherOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| H A D | custom_call_encoding.cc | 126 ShapedType shaped_type = dense.getType(); in Flatten() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 291 auto shaped_type = type.cast<mlir::ShapedType>(); in ComputeMaxUnrollFactor() local
|