Searched refs:missing_ctx_input_prefix (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 224 int missing_ctx_input_prefix, in PopulateInputs() argument
234 CHECK_GE(arg_num, missing_ctx_input_prefix); in PopulateInputs()
249 : &(ctx->input(arg_num - missing_ctx_input_prefix)); in PopulateInputs()
302 int output_num, OpKernelContext* ctx, int missing_ctx_input_prefix, in GetOrCreateTensorForOutput() argument
317 input_mapping[alias->parameter_number] - missing_ctx_input_prefix; in GetOrCreateTensorForOutput()
321 : *resource_vars_snapshots.at(missing_ctx_input_prefix + tf_param); in GetOrCreateTensorForOutput()
401 int missing_ctx_input_prefix) { in GatherVariableInfo() argument
407 int actual_input_index = write.input_index - missing_ctx_input_prefix; in GatherVariableInfo()
423 ScopedShapedBuffer output, int missing_ctx_input_prefix, in PopulateOutputs() argument
504 compilation_result->outputs[i].input_index - missing_ctx_input_prefix; in PopulateOutputs()
[all …]
|
| D | xla_launch_util.h | 88 int missing_ctx_input_prefix);
160 int missing_ctx_input_prefix,
176 xla::ScopedShapedBuffer output, int missing_ctx_input_prefix,
|