- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for CompileSerializedMlirToXlaHlo (0.43 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
mlir::DialectRegistry mlir_registry; RegisterAllTensorFlowDialects(mlir_registry); std::vector<tensorflow::TensorShape> arg_shapes; XlaCompilationResult compilation_result; auto status = CompileSerializedMlirToXlaHlo( kMlirModuleStr, arg_shapes, /*device_type=*/"XLA_TPU_JIT", /*use_tuple_args=*/true, /*enable_op_fallback=*/false, /*shape_determination_fns=*/{}, &compilation_result);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 19:54:38 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_pass_instrumentation_test.cc
} })"; SetPassThatChangedIdentity(""); std::vector<::tensorflow::TensorShape> arg_shapes = {{1}}; auto compilation_result = tensorflow::XlaCompilationResult(); TF_EXPECT_OK(tensorflow::CompileSerializedMlirToXlaHlo( legalization, arg_shapes, /*device_type=*/"XLA_TPU_JIT", /*use_tuple_args=*/true, /*enable_op_fallback=*/false,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 19 22:54:26 UTC 2023 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc
if (!mlir::SetTPUInfeedLayout(mlir_module)) return errors::Internal("Failed to set layouts attribute"); TF_ASSIGN_OR_RETURN( auto compiled_mlir, CompileSerializedMlirToXlaHlo( SerializeMlirModule(mlir_module.get()), arg_shapes, device_type, use_tuple_args, true, shape_determination_fns, compilation_result, custom_legalization_passes, metadata.module_name(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6.1K bytes - Viewed (0)