- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for EncodePrimitiveTypeAsDataType (0.21 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
const xla::Shape& xla_shape) -> absl::StatusOr<xla::Shape> { TensorShape shape; TF_RETURN_IF_ERROR(XLAShapeToTensorShape(xla_shape, &shape)); TF_ASSIGN_OR_RETURN(DataType dtype, EncodePrimitiveTypeAsDataType( xla_shape.element_type())); auto layout_preference = shape_determination_fns.layout_preference_fn( shape, dtype, std::nullopt);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
const xla::Literal* const xla_literal = arguments[i]; tensorflow::TensorShape shape; TF_ASSIGN_OR_RETURN(auto data_type, tensorflow::EncodePrimitiveTypeAsDataType( xla_literal->shape().element_type())); TF_RETURN_IF_ERROR( tensorflow::XLAShapeToTensorShape(xla_literal->shape(), &shape));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0)