Home
last modified time | relevance | path

Searched refs:SetShapeAttribute (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.h67 Status SetShapeAttribute(absl::string_view name, mlir::ShapedType shape,
Dexport_utils.cc414 Status SetShapeAttribute(absl::string_view name, mlir::ShapedType shaped_type, in SetShapeAttribute() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_tf_dialect_op.cc70 Status SetShapeAttribute(absl::string_view name, ContainerT shapes, in SetShapeAttribute() function
191 SetShapeAttribute("_output_shapes", output_shapes, attributes)); in PopulateDerivedAttributes()
Dexport_graphdef.cc223 TF_RETURN_IF_ERROR(SetShapeAttribute("_output_shapes", in GetArgumentNode()