Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h488 const std::unordered_map<string, Node*> outside_compilation_inputs,
496 const std::unordered_map<string, Node*> outside_compilation_inputs,
Ddistributed_tpu_rewrite_pass.cc3560 const std::unordered_map<string, Node*> outside_compilation_inputs, in CopyOutsideCompilationEdges() argument
3603 outside_compilation_inputs.find(outside_compilation_cluster); in CopyOutsideCompilationEdges()
3604 TF_RET_CHECK(input_iter != outside_compilation_inputs.end()); in CopyOutsideCompilationEdges()
3625 outside_compilation_inputs.find(outside_compilation_input_attr); in CopyOutsideCompilationEdges()
3626 TF_RET_CHECK(input_iter != outside_compilation_inputs.end()); in CopyOutsideCompilationEdges()
3688 const std::unordered_map<string, Node*> outside_compilation_inputs, in ReplicateOutsideCompilationEdges() argument
3693 outside_compilation_inputs, graph)); in ReplicateOutsideCompilationEdges()
4494 std::unordered_map<string, Node*> outside_compilation_inputs; in Run() local
4501 outside_compilation_inputs[lifted_arg_inputs_attr] = n; in Run()
4507 outside_compilation_inputs, graph)); in Run()