Searched refs:need_rewrite (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 56 bool* need_rewrite, int* resource_input_count, in InputTypesNeedsRearrange() argument
68 *need_rewrite = false; in InputTypesNeedsRearrange()
72 *need_rewrite = false; in InputTypesNeedsRearrange()
75 *need_rewrite = true; in InputTypesNeedsRearrange()
79 if (!*need_rewrite) { in InputTypesNeedsRearrange()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 372 bool need_rewrite = false; in RewriteDynamicReshapeSplitInput() local
385 need_rewrite = true; in RewriteDynamicReshapeSplitInput()
389 if (!need_rewrite) { in RewriteDynamicReshapeSplitInput()
584 bool need_rewrite = false; in RewriteDynamicReshapeCombineInput() local
597 need_rewrite = true; in RewriteDynamicReshapeCombineInput()
601 if (!need_rewrite) { in RewriteDynamicReshapeCombineInput()
|
| D | dynamic_dimension_inference.cc | 1264 bool need_rewrite = false; in HandleConditional() local
1312 need_rewrite = true; in HandleConditional()
1358 need_rewrite = true; in HandleConditional()
1368 if (!need_rewrite) { in HandleConditional()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 3432 bool need_rewrite, int64 id, in BuildDynamicInferenceGraph()
3459 if (!need_rewrite) { in BuildDynamicInferenceGraph()
3614 explicit WorkItem(int64 handle, bool need_rewrite) in BuildDynamicInferenceGraph()
3615 : handle(handle), need_rewrite(need_rewrite), visited(false) {} in BuildDynamicInferenceGraph()
3623 bool need_rewrite; in BuildDynamicInferenceGraph() member
3641 auto item_key = std::make_pair(item.handle, item.need_rewrite); in BuildDynamicInferenceGraph()
3698 item.need_rewrite = true; in BuildDynamicInferenceGraph()
3705 TF_RETURN_IF_ERROR(process_instruction(instr_proto, item.need_rewrite, in BuildDynamicInferenceGraph()
3716 item.need_rewrite); in BuildDynamicInferenceGraph()
3718 next_item.need_rewrite = false; in BuildDynamicInferenceGraph()
[all …]
|