/external/tensorflow/tensorflow/core/graph/ |
D | while_context.h | 38 std::vector<Node*> exit_nodes, OutputTensor cond_output, 39 std::vector<OutputTensor> body_inputs, 40 std::vector<OutputTensor> body_outputs); 45 const OutputTensor& cond_output() const { return cond_output_; } in cond_output() 46 const std::vector<OutputTensor>& body_inputs() const { return body_inputs_; } in body_inputs() 47 const std::vector<OutputTensor>& body_outputs() const { in body_outputs() 67 const OutputTensor cond_output_; 70 const std::vector<OutputTensor> body_inputs_; 71 const std::vector<OutputTensor> body_outputs_;
|
D | while_context.cc | 23 OutputTensor cond_output, in WhileContext() 24 std::vector<OutputTensor> body_inputs, in WhileContext() 25 std::vector<OutputTensor> body_outputs) in WhileContext()
|
D | graph.h | 62 struct OutputTensor; 204 Status input_tensor(int idx, OutputTensor* t) const; 336 struct OutputTensor { struct 340 OutputTensor(Node* n, int i) : node(n), index(i) {} in OutputTensor() function 341 OutputTensor() : node(nullptr), index(0) {} in OutputTensor() function 345 bool operator==(const OutputTensor& other) const; 350 uint64 operator()(OutputTensor const& s) const; argument 646 OutputTensor cond_output, 647 std::vector<OutputTensor> body_inputs, 648 std::vector<OutputTensor> body_outputs,
|
D | graph.cc | 307 Status Node::input_tensor(int idx, OutputTensor* t) const { in input_tensor() 311 *t = OutputTensor(e->src(), e->src_output()); in input_tensor() 338 bool OutputTensor::operator==(const OutputTensor& other) const { in operator ==() 342 uint64 OutputTensor::Hash::operator()(OutputTensor const& s) const { in operator ()() 795 OutputTensor cond_output, in AddWhileContext() 796 std::vector<OutputTensor> body_inputs, in AddWhileContext() 797 std::vector<OutputTensor> body_outputs, in AddWhileContext()
|
D | node_builder.h | 53 NodeOut(OutputTensor t);
|
D | node_builder.cc | 32 NodeBuilder::NodeOut::NodeOut(OutputTensor t) : NodeOut(t.node, t.index) {} in NodeOut()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | functionalize_cond.h | 61 OutputTensor output_tensor; 87 bool operator()(const OutputTensor& lhs, const OutputTensor& rhs) const; 92 using CondState = std::map<OutputTensor, BranchType, OutputTensorLess>; 125 BranchType FindBranchOf(CondId id, OutputTensor predicate) const; 188 const OutputTensor& predicate); 260 std::unordered_map<Node*, OutputTensor> merge_to_predicate_; 263 std::unordered_map<Node*, OutputTensor> merge_to_replacement_;
|
D | functionalize_cond.cc | 66 typedef std::tuple<StateMap::CondId, StateMap::AncestorId, OutputTensor> 84 string DebugString(const OutputTensor& tensor) { in DebugString() 108 const OutputTensor& pred = pred_branch.first; in DebugString() 120 Status GetSwitchPredicate(const Node& switch_node, OutputTensor* pred) { in GetSwitchPredicate() 128 *pred = OutputTensor(pred_edge->src(), pred_edge->src_output()); in GetSwitchPredicate() 132 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue() 135 *val = OutputTensor(val_edge->src(), val_edge->src_output()); in GetSwitchValue() 139 bool StateMap::OutputTensorLess::operator()(const OutputTensor& lhs, in operator ()() 140 const OutputTensor& rhs) const { in operator ()() 162 {std::make_pair(OutputTensor(nullptr, -1), BranchType::kNeither)}); in StateMap() [all …]
|
D | functionalize_cond_test.cc | 79 ss.insert(std::make_pair(OutputTensor(pred, 0), BranchType::kThenBranch)); in TEST_F() 85 ss.insert(std::make_pair(OutputTensor(pred, 0), BranchType::kElseBranch)); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | clone_constants_for_better_clustering_test.cc | 81 OutputTensor tr0_perm; in TEST() 84 OutputTensor tr1_perm; in TEST() 107 OutputTensor tr0_perm; in TEST() 110 OutputTensor tr1_perm; in TEST() 136 OutputTensor tr0_perm; in TEST() 139 OutputTensor tr1_perm; in TEST() 167 OutputTensor tr0_perm; in TEST() 170 OutputTensor tr1_perm; in TEST()
|
D | encapsulate_subgraphs_pass.cc | 130 uint64 operator()(std::pair<OutputTensor, InputTensor> const& s) const { in operator ()() 131 return Hash64Combine(OutputTensor::Hash()(s.first), in operator ()() 369 std::unordered_map<OutputTensor, int, OutputTensor::Hash> inputs; 391 std::unordered_map<OutputTensor, ArgNumAndType, OutputTensor::Hash> 479 std::unordered_map<OutputTensor, int, OutputTensor::Hash> args_by_src_; 486 std::unordered_map<OutputTensor, int, OutputTensor::Hash> results_; 583 std::unordered_set<std::pair<OutputTensor, InputTensor>, 720 return results_.at(OutputTensor(edge->src(), edge->src_output())); in GetResultIndexForEdge() 732 .inputs.at(OutputTensor(edge->src(), edge->src_output())); in GetRecvAtHostSlot() 775 std::unordered_map<OutputTensor, int, OutputTensor::Hash>::iterator iter; in RecordArg() [all …]
|
D | encapsulate_subgraphs_pass.h | 43 const std::vector<OutputTensor>& arg_source_tensors,
|
D | extract_outside_compilation_pass.h | 54 Status operator()(const std::vector<OutputTensor>&,
|
D | extract_outside_compilation_pass_test.cc | 61 std::vector<OutputTensor> arg_source_tensors; in TEST() 132 std::vector<OutputTensor> arg_source_tensors; in TEST() 158 std::vector<OutputTensor> arg_source_tensors; in TEST() 182 std::vector<OutputTensor> arg_source_tensors; in TEST() 214 std::vector<OutputTensor> arg_source_tensors; in TEST()
|
D | encapsulate_xla_computations_pass.cc | 113 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph()
|
D | encapsulate_subgraphs_pass_test.cc | 802 [&guaranteed_consts](const std::vector<OutputTensor>& arg_source_tensors, in TEST() 847 [&guaranteed_consts](const std::vector<OutputTensor>& arg_source_tensors, in TEST()
|
/external/tensorflow/tensorflow/cc/ops/ |
D | while_loop.cc | 30 OutputTensor ToOutputTensor(const Output& output) { in ToOutputTensor() 31 return OutputTensor(output.node(), output.index()); in ToOutputTensor() 35 std::vector<OutputTensor> ToOutputTensors(const std::vector<Output>& outputs) { in ToOutputTensors() 36 std::vector<OutputTensor> result(outputs.size()); in ToOutputTensors()
|
/external/tensorflow/tensorflow/c/ |
D | c_api_function.cc | 302 const std::vector<OutputTensor>& inputs, in GraphToFunctionDef() 303 const std::vector<OutputTensor>& outputs, in GraphToFunctionDef() 462 const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, in ProcessInputs() 499 std::vector<OutputTensor>* output_tensors) in ProcessOutputs() 571 std::vector<tensorflow::OutputTensor> input_tensors; in TF_GraphToFunctionWithControlOutputs() 578 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunctionWithControlOutputs()
|
D | c_api.cc | 87 using tensorflow::OutputTensor; 2805 OutputTensor tensor(&output.oper->node, output.index); in TF_TryEvaluateConstant()
|
/external/tensorflow/tensorflow/cc/framework/ |
D | while_gradients.cc | 31 Output ToOutput(OutputTensor output_tensor) { in ToOutput() 36 const std::vector<OutputTensor>& output_tensors) { in ToOutputVector()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | eval_const_tensor.h | 57 OutputTensor tensor, const ShapeRefiner& refiner,
|
D | lower_if_op.cc | 79 OutputTensor pred_;
|
D | eval_const_tensor.cc | 308 Status EvaluateConstantTensor(OutputTensor tensor, const ShapeRefiner& refiner, in EvaluateConstantTensor()
|
D | shape_refiner.cc | 422 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm.cc | 654 enum OutputTensor { enum
|