- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetConstantInputIndicesFromContext (0.71 sec)
-
tensorflow/compiler/jit/xla_launch_util_test.cc
variables_indices, &variables)); TF_ASSERT_OK_AND_ASSIGN(std::vector<int> constant_input_indices, GetConstantInputIndicesFromContext(context_.get())); TF_ASSERT_OK(LockVariables(absl::MakeSpan(variables))); TF_ASSERT_OK_AND_ASSIGN( std::vector<XlaCompiler::Argument> args, XlaComputationLaunchContext::BuildXlaCompilerArguments(
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
OP_REQUIRES(ctx, ctx->function_library(), errors::Internal("Function library missing")); // Get constants, inputs and variables from the OpKernelContext. auto constant_indices_or = GetConstantInputIndicesFromContext(ctx); OP_REQUIRES_OK(ctx, constant_indices_or.status()); std::vector<const Tensor*> inputs = InputsFromContext(ctx); std::vector<int> variable_indices = GetResourceVariableIndicesFromContext(ctx);
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/xla_launch_util.h
int missing_ctx_input_prefix); // Returns pointers to inputs stored in `ctx`. std::vector<const Tensor*> InputsFromContext(OpKernelContext* ctx); absl::StatusOr<std::vector<int>> GetConstantInputIndicesFromContext( OpKernelContext* ctx); Status SetOutputForConstant( OpKernelContext* ctx, bool requires_copy_to_device, const XlaCompiler::CompilationResult* compilation_result, int output_num);
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
for (int input_idx = 0; input_idx < ctx->num_inputs(); input_idx++) { inputs.push_back(&ctx->input(input_idx)); } return inputs; } absl::StatusOr<std::vector<int>> GetConstantInputIndicesFromContext( OpKernelContext* ctx) { std::vector<int> constant_input_indices; TF_RETURN_IF_ERROR(GetCompileTimeConstInputs( &ctx->op_kernel(), &constant_input_indices, ctx->function_library()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0)