| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | attr_util.cc | 183 auto shape_attr = list_attr.GetAttributeOfType<tfrt::ShapeAttr>(i); in FillAttrValueMapUsingAggregate() local
228 auto shape_attr = attrs.GetAsserting<tfrt::ShapeAttr>(entry.name); in FillAttrValueMapUsingScalar() local
549 if (auto shape_attr = bef_attr.dyn_cast<tfrt::ShapeAttr>()) { in SetUpScalarAttr() local
607 if (auto shape_attr = base.dyn_cast<tfrt::ShapeAttr>()) { in SetUpListAttr() local
|
| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | host_compute_ops.cc | 110 const AttrValue* shape_attr; in __anon6ab076a10402() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | function_ops.cc | 37 const AttrValue* shape_attr = context->GetAttr("_output_shapes"); in __anon3d55e0050102() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 582 TensorShape shape_attr; in __anon592c2f360402() local
1077 TensorShape shape_attr; in __anon592c2f361002() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | corert_converter.cc | 313 if (auto shape_attr = attr.dyn_cast<mlir::TF::ShapeAttr>()) { in ConvertAttribute() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_constants_test.cc | 145 tensorflow::AttrValue shape_attr; in TestPreserveOutputShapes() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 353 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in legalizeFilter() local
426 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in RewriteNewAxisMask() local
|
| D | optimize.cc | 344 DenseIntElementsAttr shape_attr = in ShapeMatchesReduceWithKeepAxes() local
|
| D | lower_static_tensor_list.cc | 446 SmallVector<Attribute, 4> shape_attr; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 448 tfrt::ShapeAttr shape_attr = in PrepareAttributes() local
520 auto shape_attr = list_attr.GetAttributeOfType<ShapeAttr>(i); in PrepareAttributes() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 1511 AttrValue shape_attr; in TEST() local
1535 const auto& shape_attr = attr.list().shape(0); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 335 auto shape_attr = DenseIntElementsAttr::get(type, output_ty.getShape()); in matchAndRewrite() local
721 DenseIntElementsAttr shape_attr; in GetReshapeOutputType() local
2085 DenseIntElementsAttr shape_attr; in GetSlicedShapeAndBoundRanges() local
|
| D | tf_ops_a_m.cc | 1394 if (auto shape_attr = shape.dyn_cast_or_null<DenseIntElementsAttr>()) in fold() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.cc | 455 auto shape_attr = TF::ShapeAttr::get(rewriter.getContext(), shape); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 222 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_output_shape); in convertPackOp() local
1187 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertExpandDimsOp() local
1247 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertSqueezeOp() local
|
| D | legalize_tf.cc | 673 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite() local
723 auto shape_attr = in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1439 ElementsAttr shape_attr; in matchAndRewrite() local
1512 auto shape_attr = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 859 mlir::DenseIntElementsAttr shape_attr; in ConvertOp() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt.cc | 1635 tfrt::ShapeAttr shape_attr(buf.data() + offset); in SetAttrShape() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 853 AttrValue shape_attr; in UpdateFunction() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 2414 AttrValue shape_attr; in CreateIdentityGraphDef() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1623 DenseIntElementsAttr shape_attr; in GetReshapeOutputType() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1135 const AttrValue* shape_attr = node.attrs().Find("_handle_shapes"); in InferOutputType() local
|