- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetTensorShapeFromXlaArgument (0.18 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
"tf2xla.collective_info.group_key"; // Extracts shape from XlaArgument as TensorShape. If shape is a xla::Shape, // that is converted to a TensorShape. absl::StatusOr<TensorShape> GetTensorShapeFromXlaArgument( const XlaArgument& arg) { if (absl::holds_alternative<xla::Shape>(arg.shape)) { TensorShape arg_shape; TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0)