| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| H A D | attr_util.cc | 182 auto shape_attr = list_attr.GetAttributeOfType<tfrt::ShapeAttr>(i); in FillAttrValueMapUsingAggregate() local
227 auto shape_attr = attrs.GetAsserting<tfrt::ShapeAttr>(entry.name); in FillAttrValueMapUsingScalar() local
548 if (auto shape_attr = bef_attr.dyn_cast<tfrt::ShapeAttr>()) { in SetUpScalarAttr() local
606 if (auto shape_attr = base.dyn_cast<tfrt::ShapeAttr>()) { in SetUpListAttr() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| H A D | xla_ops.cc | 585 TensorShape shape_attr; in __anon6b9e6b800402() local
1090 TensorShape shape_attr; in __anon6b9e6b801002() local
1256 TensorShape shape_attr; in __anon6b9e6b801202() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ops/ |
| H A D | host_compute_ops.cc | 108 const AttrValue* shape_attr; in __anone2a2ed6b0402() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | function_ops.cc | 37 const AttrValue* shape_attr = context->GetAttr("_output_shapes"); in __anon748b2ccf0102() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | prepare_tf.cc | 395 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in legalizeFilter() local
468 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in RewriteNewAxisMask() local
|
| H A D | optimize.cc | 334 DenseIntElementsAttr shape_attr = in ShapeMatchesReduceWithKeepAxes() local
|
| H A D | lower_static_tensor_list.cc | 508 SmallVector<Attribute, 4> shape_attr; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | shape_utils.cc | 103 auto shape_attr = global_shape_attr[output_index]; in ExtractGlobalOutputShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_kernels.cc | 450 tfrt::ShapeAttr shape_attr = in PrepareAttributes() local
522 auto shape_attr = list_attr.GetAttributeOfType<ShapeAttr>(i); in PrepareAttributes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/ |
| H A D | shape_inference_utils.cc | 112 auto shape_attr = attrs[0].dyn_cast<tf_type::ShapeAttr>(); in GetShapeFromMlirAttr() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | function_test.cc | 1574 AttrValue shape_attr; in TEST() local
1598 const auto& shape_attr = attr.list().shape(0); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | corert_converter.cc | 317 if (auto shape_attr = attr.dyn_cast<mlir::TF::ShapeAttr>()) { in ConvertAttribute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | fold_constants_test.cc | 145 tensorflow::AttrValue shape_attr; in TestPreserveOutputShapes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | save_restore_spmd_expander.cc | 680 mlir::TF::ShapeAttr shape_attr = shape.cast<mlir::TF::ShapeAttr>(); in ExpandDTensorRestoreV2Op() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_common.cc | 255 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_output_shape); in convertPackOp() local
1235 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertExpandDimsOp() local
1298 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertSqueezeOp() local
|
| H A D | legalize_tf.cc | 674 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite() local
724 auto shape_attr = in matchAndRewrite() local
|
| H A D | legalize_tfl.cc | 1641 ElementsAttr shape_attr; in matchAndRewrite() local
1713 auto shape_attr = in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 352 auto shape_attr = DenseIntElementsAttr::get(type, output_ty.getShape()); in matchAndRewrite() local
760 DenseIntElementsAttr shape_attr; in GetReshapeOutputType() local
2171 DenseIntElementsAttr shape_attr; in GetSlicedShapeAndBoundRanges() local
|
| H A D | tf_ops_a_m.cc | 1466 if (auto shape_attr = shape.dyn_cast_or_null<DenseIntElementsAttr>()) in fold() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| H A D | tfr_ops.cc | 486 auto shape_attr = TF::ShapeAttr::get(rewriter.getContext(), shape); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | flatbuffer_import.cc | 888 mlir::DenseIntElementsAttr shape_attr; in ConvertOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 1701 tfrt::ShapeAttr shape_attr(buf.data() + offset); in SetAttrShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 403 Type GetType(Attribute shape_attr, Attribute type_attr) { in GetType()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 859 AttrValue shape_attr; in UpdateFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 2123 auto shape_attr = shape_op->getAttrOfType<ElementsAttr>("value"); in matchAndRewrite() local
|