Searched refs:input_name_map (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 530 Graph* graph, NameRangeMap* input_name_map, const NameAttrList** function,
536 const Node& replicate_node, const NameRangeMap& input_name_map,
|
| D | distributed_tpu_rewrite_pass.cc | 3973 Graph* graph, NameRangeMap* input_name_map, const NameAttrList** function, in GetIOTypes() argument
4002 input_name_map, &output_name_map)); in GetIOTypes()
4097 const Node& replicate_node, const NameRangeMap& input_name_map, in DealWithConstantsAndVariables() argument
4103 replicate_node, input_name_map, guaranteed_constant_nodes)); in DealWithConstantsAndVariables()
4243 NameRangeMap input_name_map; in RewriteTPUReplicateNode() local
4249 &input_name_map, &function, &computation, in RewriteTPUReplicateNode()
4303 FindVariableInputs(*replicate_node, input_name_map, &variable_inputs)); in RewriteTPUReplicateNode()
4308 *replicate_node, input_name_map, graph, host_transfer_sequencer, in RewriteTPUReplicateNode()
|