- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for RunPjRtExecutable (0.88 sec)
-
tensorflow/compiler/jit/xla_launch_util.h
// `variables` are the input arguments to the computation, usually read from the // OpKernelContext, `ctx`. Requires the device-appropriate `pjrt_client` and the // `compilation_result` used to build the `executable`. Status RunPjRtExecutable( const std::vector<const Tensor*>& inputs, const std::vector<VariableInfo>& variables, const XlaCompiler::CompilationResult& compilation_result,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
} return RunPjRtExecutable(/*num_missing_prefix_ctx_inputs=*/0, inputs, variable_snapshots, variables, compilation_result, pjrt_client, executable, ctx); } // TODO(b/289421064): Add unit test for this. Status RunPjRtExecutable( int num_missing_prefix_ctx_inputs, const std::vector<const Tensor*>& inputs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util_test.cc
EXPECT_TRUE(options.untuple_result); EXPECT_FALSE(options.strict_shape_checking); EXPECT_TRUE(options.use_major_to_minor_data_layout_for_callbacks); } TEST_F(PjRtExecutionUtilTest, RunPjRtExecutable) { XlaOpRegistry::RegisterCompilationKernels(); TF_EXPECT_OK(NodeDefBuilder("AddV2", "AddV2") .Input(FakeInput(DT_INT32)) .Input(FakeInput(DT_INT32))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_on_demand_op.cc
VLOG(2) << "result != nullptr: " << (result != nullptr); VLOG(2) << "pjrt_executable != nullptr: " << (pjrt_executable != nullptr); VLOG(2) << "Executing with PJRT ..."; OP_REQUIRES_OK(ctx, RunPjRtExecutable(inputs, variables, *result, pjrt_device_compiler->client(), pjrt_executable, ctx)); VLOG(2) << "Completed executing with PJRT!";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 08:39:39 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
done); OP_REQUIRES_OK_ASYNC(ctx, LockVariables(absl::MakeSpan(variable_infos)), done); OP_REQUIRES_OK_ASYNC( ctx, RunPjRtExecutable(inputs, variable_infos, *compilation_result, pjrt_client, pjrt_executable, ctx), done); } VLOG(2) << "Done executing with PJRT."; done(); };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0)