Home
last modified time | relevance | path

Searched refs:num_constant_args (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc80 ResourceVarsSnapshot resource_var_snapshots, int num_constant_args) in XlaExecutableClosure() argument
85 num_constant_args_(num_constant_args) {} in XlaExecutableClosure()
98 int num_constant_args() const { return num_constant_args_; } in num_constant_args() function in tensorflow::__anon46563ba90111::XlaExecutableClosure
502 /*missing_ctx_input_prefix=*/closure.num_constant_args(), in Compute()
536 ctx, *closure.compilation_result(), closure.num_constant_args()); in Compute()
543 /*missing_ctx_input_prefix=*/closure.num_constant_args(), in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc89 const string& node_name, int num_constant_args, in MakeXlaCompiledKernel() argument
95 AddNodeAttr(kXlaNumConstantArgsAttr, num_constant_args, &call_node); in MakeXlaCompiledKernel()