- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for Dtype (0.02 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
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); } return absl::OkStatus(); } absl::Status SetupReturnValues(mlir::ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0)