- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ConvertToDataType (0.17 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
TF_RETURN_IF_ERROR(tensorflow::TensorShape::BuildTensorShape( xla_shape.dimensions(), &tensor_shape)); arg_shapes.emplace_back(tensor_shape); DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(input_type, &dtype)); auto metadata_arg = metadata_proto.add_args(); metadata_arg->set_kind(tpu::TPUCompileMetadataProto::Arg::PARAMETER); metadata_arg->set_dtype(dtype); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (!subtypes.empty()) { AttrValue handle_dtypes_attr; AttrValue handle_shapes_attr; for (mlir::TensorType subtype : subtypes) { DataType dtype; TF_RETURN_IF_ERROR(ConvertToDataType(subtype.getElementType(), &dtype)); handle_dtypes_attr.mutable_list()->add_type(dtype); SetTensorShapeProto(subtype, handle_shapes_attr.mutable_list()->add_shape());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0)