| /external/tensorflow/tensorflow/core/tpu/ops/ |
| D | host_compute_ops.cc | 104 const AttrValue* shape_attr; in __anonc8f95cfc0402() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | function_ops.cc | 37 const AttrValue* shape_attr = context->attrs().Find("_output_shapes"); in __anon5bd79b600102() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| D | xla_ops.cc | 509 TensorShape shape_attr; in __anonbbc1c2d10402() local
884 TensorShape shape_attr; in __anonbbc1c2d10d02() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.cc | 345 if (auto shape_attr = attr.dyn_cast<ShapeAttr>()) in printAttribute() local
|
| D | tf_ops_n_z.cc | 590 DenseIntElementsAttr shape_attr; in GetReshapeOutputType() local
2030 DenseIntElementsAttr shape_attr; in GetSlicedShapeAndBoundRanges() local
|
| D | tf_ops_a_m.cc | 1376 if (auto shape_attr = shape.dyn_cast_or_null<DenseIntElementsAttr>()) in fold() 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 | 498 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in legalizeFilter() local
564 auto shape_attr = DenseElementsAttr::get(shape_type, result_shape_data); in RewriteNewAxisMask() local
|
| D | optimize.cc | 270 DenseIntElementsAttr shape_attr = in ShapeMatchesReduceWithKeepAxes() local
|
| D | lower_static_tensor_list.cc | 374 SmallVector<Attribute, 4> shape_attr; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.cc | 427 auto shape_attr = TF::ShapeAttr::get(rewriter.getContext(), shape); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 1507 AttrValue shape_attr; in TEST() local
1531 const auto& shape_attr = attr.list().shape(0); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 200 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_output_shape); in convertPackOp() local
1281 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertExpandDimsOp() local
1341 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(reshape_dims); in convertSqueezeOp() local
|
| D | legalize_tf.cc | 669 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite() local
719 auto shape_attr = DenseElementsAttr::get( in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1587 ArrayAttr shape_attr = rewriter.getI64ArrayAttr(shape_vals); in matchAndRewrite() local
1637 auto shape_attr = DenseElementsAttr::get( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 729 mlir::DenseIntElementsAttr shape_attr; in ConvertOp() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 830 AttrValue shape_attr; in UpdateFunction() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1362 DenseIntElementsAttr shape_attr; in GetReshapeOutputType() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 2407 AttrValue shape_attr; in CreateIdentityGraphDef() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1040 const AttrValue* shape_attr = node.attrs().Find("_handle_shapes"); in InferOutputType() local
|