- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 60 for compilation_result (0.39 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
return %cst : tensor<2xi32> } })"; auto compilation_result = CompileMlirModule( kHasReturnValuesAndNoMetadataRetvals, ConfigProto::Experimental::MLIR_BRIDGE_ROLLOUT_UNSPECIFIED); EXPECT_TRUE(compilation_result.ok()); // Ensure that the compilation result contains a constant. EXPECT_THAT(compilation_result, ComputationProtoContains("opcode:.*constant")); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc
return %0 : tensor<32x64xf32> } })"; auto compilation_result = CompileWithComputation(CreateTestMlirToHloArgs(kHasReturnValues)); EXPECT_TRUE(compilation_result.ok()); // The StatelessRandomNormal must not be replaced by a literal tensor. EXPECT_THAT(compilation_result, ComputationProtoContains("tf.StatelessRandomNormal")); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc
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); EXPECT_TRUE(status.ok()); EXPECT_THAT(status.value(), HasSubstr("mhlo.const")); }
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/api/v1/compile_tf_graph.cc
GuaranteedConsts consts; *compilation_result = {}; TF_RETURN_IF_ERROR(CompileTFFunctionToHlo( *flib_def, versions.producer(), shape_determination_funcs, arg_shapes, consts, func, metadata, client, arg_core_mapping, per_core_arg_shapes, use_tuple_args, compilation_result)); return PopulateInputOutputAliasing(main_fn, compilation_result, use_tuple_args);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc
shape_determination_fns, use_tuple_args, compilation_result, custom_legalization_passes, arg_shapes, arg_core_mapping, per_core_arg_shapes); if (mlir_bridge_status.ok()) { VLOG(1) << "Successfully compiled MLIR computation to XLA HLO using MLIR " "tf2xla bridge"; return *compilation_result; } tsl::error_logging::Log(kBridgeComponent,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.h
const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, XlaCompilationResult* compilation_result); // Runs MLIR Bridge on a MLIR module. // // If lower_to_xla_hlo is true then compiles down into XLA HLO, generates all // accompanying metadata and stores them in CompilationResult. // // If enable_op_fallback is set to false, graph is legalized only if the graph
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/jit/device_executable_persistor.h
uint64 signature_hash, const XlaCompiler::Options& options, const XlaCompiler::CompilationResult& compilation_result, const ExecutableType& executable, DeviceCompilerClient<ExecutableType, ClientType>* compiler_client) const { XlaSerializedCacheEntry serialized_entry; const xla::HloModuleProto& hlo_module = compilation_result.computation->proto(); *serialized_entry.mutable_key() =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
namespace { mlir::LogicalResult PrintHloModuleText( const XlaCompilationResult& compilation_result, llvm::raw_ostream& output) { const xla::HloModuleConfig module_config( compilation_result.computation->GetProgramShape().value()); auto status_or_hlo_module = xla::HloModule::CreateFromProto( compilation_result.computation->proto(), module_config); if (!status_or_hlo_module.ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir_test.cc
std::vector<std::unique_ptr<mlir::Pass>> custom_legalization_passes; auto compilation_result = std::make_unique<XlaCompilationResult>(); return CompileFromMlirToXlaHlo( compile_to_xla_hlo, mlir_to_hlo_args, metadata_proto, /*device_type=*/"XLA_TPU_JIT", /*shape_determination_fns=*/{}, use_tuple_args, compilation_result.get(), custom_legalization_passes, arg_shapes, &arg_core_mapping,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_compile_util.h
const NameAttrList& function, const std::vector<XlaCompiler::Argument>& args, DeviceCompileMode compile_mode, bool has_ref_vars, bool may_alias_resource_update, const XlaCompiler::CompilationResult** compilation_result, xla::PjRtClient** client, xla::PjRtLoadedExecutable** executable); // Similar to the above function but it does not take a OpKernelContext. // Instead, it takes the following arguments that are obtained from
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K bytes - Viewed (0)