- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for graph_def_version (0.32 sec)
-
tensorflow/compiler/jit/xla_compiler_options_util.cc
<< ",device_ordinal=" << options.device_ordinal << ",client=" << options.client << ",flib_def=" << options.flib_def << ",graph_def_version=" << options.graph_def_version << ",options.shape_determination_fns.layout_preference_fn?=" << (options.shape_determination_fns.layout_preference_fn != nullptr)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util_test.cc
/*pjrt_device_compiler=*/nullptr); EXPECT_EQ(options.device_type, compilation_device_type); EXPECT_EQ(options.device_ordinal, 0); EXPECT_NE(options.flib_def, nullptr); EXPECT_EQ(options.graph_def_version, TF_GRAPH_DEF_VERSION); EXPECT_FALSE(options.allow_cpu_custom_calls); EXPECT_FALSE(options.alias_passthrough_params); // Check if options have the supplied shape determination functions set.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util.h
// compilation and execution. XlaCompiler::Options GenerateCompilerOptionsForPjRt( const FunctionLibraryDefinition* function_library_def, int graph_def_version, const DeviceBase* device_base, const XlaPlatformInfo& platform_info, const DeviceCompiler<xla::PjRtLoadedExecutable, xla::PjRtClient>* pjrt_device_compiler);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator.cc
flr, dev->resource_manager(), props, input_memory_types, output_memory_types, flr->graph_def_version(), &s); *kernel = std::make_unique<XlaLocalLaunchBase>( &construction, constant_arg_indices, resource_arg_indices, function, /*has_ref_vars=*/false); return s; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/jit/get_compiler_ir.cc
} else { options.device_type = compilation_device_type; options.flib_def = flr->GetFunctionLibraryDefinition(); options.graph_def_version = flr->graph_def_version(); options.allow_cpu_custom_calls = (platform_info.platform_id() == se::host::kHostPlatformId); options.alias_passthrough_params = !platform_info.is_on_xla_device(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 19K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc
NameAttrList function; function.set_name("empty"); FunctionToHloArgs function_to_hlo_args = {&function, &flib_def, /*graph_def_version=*/0, {&guaranteed_constants}}; TF_EXPECT_OK(CompileWithComputation(function_to_hlo_args).status()); Histogram histogram =
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/v2/legalize_tf_test.cc
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.cc
xla::CompileOnlyClient* client, XlaCompiler::CompilationResult* compilation_result) { Status comp_status = CompileTFFunctionToHlo( *function_computation.flib_def, function_computation.graph_def_version, shape_determination_funcs, arg_shapes, function_computation.guaranteed_constants, *function_computation.function, metadata, client, arg_core_mapping, per_core_arg_shapes, 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/c/kernels.cc
->frame_iter() .frame_id; } int TF_GetGraphDefVersion(TF_OpKernelContext* ctx) { return reinterpret_cast<::tensorflow::OpKernelContext*>(ctx) ->function_library() ->graph_def_version(); } int64_t TF_GetIterId(TF_OpKernelContext* ctx) { return reinterpret_cast<::tensorflow::OpKernelContext*>(ctx) ->frame_iter() .iter_id; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)