- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SetShapeAttribute (0.39 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
template <typename ContainerT, typename = typename std::enable_if<std::is_same< std::optional<llvm::ArrayRef<int64_t>>, decltype(*std::declval<ContainerT>().begin())>::value>::type> void SetShapeAttribute(absl::string_view name, ContainerT shapes, AttrValueMap* values) { AttrValue value; auto& shape_list = *value.mutable_list(); for (const std::optional<llvm::ArrayRef<int64_t>>& shape : shapes) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
std::move(value); } } for (auto& it : func_call_attrs) { (*values)[it.first] = std::move(it.second); } return absl::OkStatus(); } Status SetShapeAttribute(absl::string_view name, mlir::ShapedType shaped_type, AttrValueMap* values) { AttrValue value; SetTensorShapeProto(shaped_type, value.mutable_list()->add_shape());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
} (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype)); AttrValue type_attr; type_attr.set_type(dtype);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} (*node_def->mutable_attr())["_handle_dtypes"] = handle_dtypes_attr; (*node_def->mutable_attr())["_handle_shapes"] = handle_shapes_attr; } } TF_RETURN_IF_ERROR( SetShapeAttribute("_output_shapes", arg_type, node_def->mutable_attr())); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(arg_type.getElementType(), &dtype)); AttrValue type_attr; type_attr.set_type(dtype);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0)