- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for data_type_ (0.08 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
xla::Shape xla_shape = xla::TypeToShape(input_type); 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);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0)