Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc56 bool* need_rewrite, int* resource_input_count, in InputTypesNeedsRearrange() argument
83 *resource_input_count = 0; in InputTypesNeedsRearrange()
87 ++(*resource_input_count); in InputTypesNeedsRearrange()
91 resource_index = in_types.size() - *resource_input_count; in InputTypesNeedsRearrange()
133 int resource_input_count, in ReorderOutputEdges() argument
149 if (new_src_output < input_count - resource_input_count) { in ReorderOutputEdges()
284 int resource_input_count; in MaybeRewriteWhileNode() local
287 types, &input_need_rearrange, &resource_input_count, &index_mapping)); in MaybeRewriteWhileNode()
304 resource_input_count, index_mapping)); in MaybeRewriteWhileNode()
355 if (new_index < types.size() - resource_input_count) { in MaybeRewriteWhileNode()
[all …]