Home
last modified time | relevance | path

Searched refs:SetTensorShapeProto (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dexport_utils.h71 void SetTensorShapeProto(ShapeContainerT shape, TensorShapeProto* proto) { in SetTensorShapeProto() function
H A Dexport_utils.cc131 SetTensorShapeProto(attr, value->mutable_shape()); in ConvertAttribute()
446 SetTensorShapeProto(shaped_type, value.mutable_list()->add_shape()); in SetShapeAttribute()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_tensor.h58 void SetTensorShapeProto(ShapeContainerT shape, in SetTensorShapeProto() function
H A Dconvert_attributes.cc98 SetTensorShapeProto(attr, value->mutable_shape()); in ConvertAttribute()
283 SetTensorShapeProto(shaped_type, value.mutable_shape()); in SetShapeAttribute()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dexport_graphdef.cc196 SetTensorShapeProto(subtype, in GetArgumentNode()