Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dimmutable_executor_state.cc88 ControlFlowInfo cf_info; in Initialize() local
89 TF_RETURN_IF_ERROR(BuildControlFlowInfo(&graph, &cf_info)); in Initialize()
91 for (auto& it : cf_info.unique_frame_names) { in Initialize()
123 const string& frame_name = cf_info.frame_names[id]; in Initialize()
259 InitializePending(&graph, cf_info); in Initialize()
284 ControlFlowInfo* cf_info) { in BuildControlFlowInfo() argument
286 cf_info->frame_names.resize(num_nodes); in BuildControlFlowInfo()
299 cf_info->unique_frame_names.insert(frame_name); in BuildControlFlowInfo()
318 frame_name = cf_info->frame_names[parent->id()]; in BuildControlFlowInfo()
322 frame_name = cf_info->frame_names[curr_id]; in BuildControlFlowInfo()
[all …]
Dimmutable_executor_state.h126 ControlFlowInfo* cf_info);
127 void InitializePending(const Graph* graph, const ControlFlowInfo& cf_info);
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc114 std::vector<ControlFlowInfo> cf_info; member
502 std::vector<ControlFlowInfo>* cf_info) { in AddControlFlowInfo() argument
504 if (static_cast<size_t>(id) >= cf_info->size()) { in AddControlFlowInfo()
505 cf_info->resize(id + 1); in AddControlFlowInfo()
507 const ControlFlowInfo& src_info = (*cf_info)[src->id()]; in AddControlFlowInfo()
508 ControlFlowInfo* info = &(*cf_info)[id]; in AddControlFlowInfo()
522 std::vector<ControlFlowInfo>* cf_info, in AddControlLoop() argument
526 const ControlFlowInfo& src_info = (*cf_info)[src->id()]; in AddControlLoop()
558 AddControlFlowInfo(enter, src, cf_info); in AddControlLoop()
559 AddControlFlowInfo(merge, src, cf_info); in AddControlLoop()
[all …]
Dcontrol_flow.cc42 const std::vector<ControlFlowInfo>& cf_info) { in ValidateControlFlowInfo() argument
45 const ControlFlowInfo& cf = cf_info[node->id()]; in ValidateControlFlowInfo()
53 Frame* parent = &frames[cf_info[cf.parent_frame->id()].frame_name]; in ValidateControlFlowInfo()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.cc53 const std::vector<ControlFlowInfo>& cf_info, const Graph* graph, in ExtractWhileLoopFrames() argument
57 const ControlFlowInfo& cf = cf_info[node->id()]; in ExtractWhileLoopFrames()
68 &(*frames)[cf_info[cf.parent_frame->id()].frame_name]; in ExtractWhileLoopFrames()
Dfunctionalize_control_flow_util.h78 const std::vector<ControlFlowInfo>& cf_info, const Graph* graph,
Dfunctionalize_while.cc517 std::vector<ControlFlowInfo> cf_info; in FunctionalizeWhileLoop() local
519 TF_RETURN_IF_ERROR(BuildControlFlowInfo(graph, &cf_info, &unreachable_nodes)); in FunctionalizeWhileLoop()
529 ExtractWhileLoopFrames(cf_info, graph, &frames, node_filter)); in FunctionalizeWhileLoop()
/external/autotest/cli/
Djob.py499 cf_info = self.execute_rpc(op='generate_control_file',
503 self.data['control_file'] = cf_info['control_file']
505 self.data['synch_count'] = cf_info['synch_count']
506 if cf_info['is_server']:
513 deps = sorted(deps.union(cf_info['dependencies']))
/external/mesa3d/src/amd/compiler/
Daco_instruction_selection.cpp5334 if (ctx->cf_info.loop_nest_depth || ctx->cf_info.parent_if.is_divergent) in visit_discard_if()
5335 ctx->cf_info.exec_potentially_empty_discard = true; in visit_discard_if()
5353 if (ctx->cf_info.loop_nest_depth || ctx->cf_info.parent_if.is_divergent) in visit_discard()
5354 ctx->cf_info.exec_potentially_empty_discard = true; in visit_discard()
5356 bool divergent = ctx->cf_info.parent_if.is_divergent || in visit_discard()
5357 ctx->cf_info.parent_loop.has_divergent_continue; in visit_discard()
5364 Block *linear_target = ctx->cf_info.parent_loop.exit; in visit_discard()
5370 ctx->cf_info.has_branch = true; in visit_discard()
5399 if (!ctx->cf_info.parent_if.is_divergent) { in visit_discard()
8308 if (ctx->cf_info.loop_nest_depth || ctx->cf_info.parent_if.is_divergent) in visit_intrinsic()
[all …]
Daco_instruction_selection.h85 } cf_info; member
Daco_instruction_selection_setup.cpp1007 ctx->cf_info.nir_to_aco.reset(nir_to_aco.release()); in init_context()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc429 std::vector<ControlFlowInfo> cf_info; in DetectHostTrainingLoop() local
431 BuildControlFlowInfo(graph, &cf_info, /*unreachable_nodes=*/nullptr)); in DetectHostTrainingLoop()
434 TF_RETURN_IF_ERROR(ExtractWhileLoopFrames(cf_info, graph, &frames)); in DetectHostTrainingLoop()
/external/autotest/frontend/afe/
Drpc_interface.py887 cf_info, test_objects, profiler_objects = (
890 cf_info['control_file'] = control_file_lib.generate_control(
892 is_server=cf_info['is_server'],
895 return cf_info
Drpc_utils.py285 cf_info = dict(is_server=is_server, synch_count=synch_count,
287 return cf_info, test_objects, profiler_objects