| /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 | 69 struct OutputTensor;
242 Status input_tensor(int idx, OutputTensor* t) const;
395 struct OutputTensor { struct
399 OutputTensor(Node* n, int i) : node(n), index(i) {} in OutputTensor() argument
400 OutputTensor() : node(nullptr), index(0) {} in OutputTensor() function
404 bool operator==(const OutputTensor& other) const;
409 uint64 operator()(OutputTensor const& s) const; argument
713 OutputTensor cond_output,
714 std::vector<OutputTensor> body_inputs,
715 std::vector<OutputTensor> body_outputs,
|
| D | graph.cc | 318 Status Node::input_tensor(int idx, OutputTensor* t) const { in input_tensor()
322 *t = OutputTensor(e->src(), e->src_output()); in input_tensor()
355 bool OutputTensor::operator==(const OutputTensor& other) const { in operator ==()
359 uint64 OutputTensor::Hash::operator()(OutputTensor const& s) const { in operator ()()
866 OutputTensor cond_output, in AddWhileContext()
867 std::vector<OutputTensor> body_inputs, in AddWhileContext()
868 std::vector<OutputTensor> body_outputs, in AddWhileContext()
|
| D | node_builder.h | 53 NodeOut(OutputTensor t);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.h | 72 OutputTensor output_tensor;
98 bool operator()(const OutputTensor& lhs, const OutputTensor& rhs) const;
103 using CondState = std::map<OutputTensor, BranchType, OutputTensorLess>;
136 BranchType FindBranchOf(CondId id, OutputTensor predicate) const;
197 const OutputTensor& predicate);
270 std::unordered_map<Node*, OutputTensor> merge_to_predicate_;
273 std::unordered_map<Node*, OutputTensor> merge_to_replacement_;
|
| D | functionalize_cond.cc | 68 typedef std::tuple<StateMap::CondId, StateMap::AncestorId, OutputTensor>
86 string DebugString(const OutputTensor& tensor) { in DebugString()
110 const OutputTensor& pred = pred_branch.first; in DebugString()
122 Status GetSwitchPredicate(const Node& switch_node, OutputTensor* pred) { in GetSwitchPredicate()
130 *pred = OutputTensor(pred_edge->src(), pred_edge->src_output()); in GetSwitchPredicate()
134 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue()
137 *val = OutputTensor(val_edge->src(), val_edge->src_output()); in GetSwitchValue()
141 bool StateMap::OutputTensorLess::operator()(const OutputTensor& lhs, in operator ()()
142 const OutputTensor& rhs) const { in operator ()()
164 {std::make_pair(OutputTensor(nullptr, -1), BranchType::kNeither)}); in StateMap()
[all …]
|
| D | functionalize_cond_test.cc | 85 ss.insert(std::make_pair(OutputTensor(pred, 0), BranchType::kThenBranch)); in TEST_F()
91 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 | 82 OutputTensor tr0_perm; in TEST()
85 OutputTensor tr1_perm; in TEST()
108 OutputTensor tr0_perm; in TEST()
111 OutputTensor tr1_perm; in TEST()
138 OutputTensor tr0_perm; in TEST()
142 OutputTensor tr1_perm; in TEST()
169 OutputTensor tr0_perm; in TEST()
172 OutputTensor tr1_perm; in TEST()
200 OutputTensor tr0_perm; in TEST()
203 OutputTensor tr1_perm; in TEST()
|
| D | encapsulate_subgraphs_pass.cc | 134 uint64 operator()(std::pair<OutputTensor, InputTensor> const& s) const { in operator ()()
135 return Hash64Combine(OutputTensor::Hash()(s.first), in operator ()()
279 std::unordered_map<OutputTensor, int, OutputTensor::Hash> args_by_src_;
286 std::unordered_map<OutputTensor, int, OutputTensor::Hash> results_;
355 std::unordered_set<std::pair<OutputTensor, InputTensor>,
439 return results_.at(OutputTensor(edge->src(), edge->src_output())); in GetResultIndexForEdge()
470 std::unordered_map<OutputTensor, int, OutputTensor::Hash>::iterator iter; in RecordArg()
473 OutputTensor(src_node, src_slot), args_by_src_.size()); in RecordArg()
515 std::unordered_map<OutputTensor, int, OutputTensor::Hash>::iterator iter; in RecordResult()
518 results_.emplace(OutputTensor(src_node, src_slot), results_.size()); in RecordResult()
[all …]
|
| D | encapsulate_subgraphs_pass.h | 52 const std::vector<OutputTensor>& arg_source_tensors,
|
| D | extract_outside_compilation_pass.h | 55 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()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.h | 36 const std::vector<OutputTensor>& inputs,
37 const std::vector<OutputTensor>& outputs,
|
| D | graph_to_functiondef.cc | 300 std::vector<OutputTensor>* args_or_retvals) { in GraphToFunctionDefHelper()
316 std::vector<OutputTensor> inputs; in GraphToFunctionDefHelper()
317 std::vector<OutputTensor> outputs; in GraphToFunctionDefHelper()
343 [](const std::vector<OutputTensor>& args_or_retvals, in GraphToFunctionDefHelper()
372 const std::vector<OutputTensor>& inputs, in GraphToFunctionDef()
373 const std::vector<OutputTensor>& outputs, in GraphToFunctionDef()
|
| /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 | 52 const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, in ProcessInputs()
89 std::vector<OutputTensor>* output_tensors) in ProcessOutputs()
161 std::vector<tensorflow::OutputTensor> input_tensors; in TF_GraphToFunctionWithControlOutputs()
168 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunctionWithControlOutputs()
|
| /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 | 60 OutputTensor tensor, const ShapeRefiner& refiner,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 240 absl::InlinedVector<OutputTensor, 4>* arg_nodes,
241 absl::InlinedVector<OutputTensor, 4>* ret_nodes,
257 const absl::InlinedVector<OutputTensor, 4>& arg_nodes,
258 const absl::InlinedVector<OutputTensor, 4>& ret_nodes,
394 const absl::InlinedVector<OutputTensor, 4>& arg_nodes,
395 const absl::InlinedVector<OutputTensor, 4>& ret_nodes,
642 absl::InlinedVector<OutputTensor, 4> arg_nodes; in ConvertDeferredFunctions()
643 absl::InlinedVector<OutputTensor, 4> ret_nodes; in ConvertDeferredFunctions()
1316 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody() argument
1317 absl::InlinedVector<OutputTensor, 4>* ret_nodes, in GetArgsAndRetsFromFunctionBody()
[all …]
|
| /external/armnn/src/armnnTfLiteParser/test/ |
| D | Constant.cpp | 58 "name": "OutputTensor", in ConstantAddFixture()
|
| D | ResizeBilinear.cpp | 58 "name": "OutputTensor", in ResizeBilinearFixture()
|
| D | ResizeNearestNeighbor.cpp | 58 "name": "OutputTensor", in ResizeNearestNeighborFixture()
|
| D | AvgPool2D.cpp | 30 "name": "OutputTensor", in AvgPool2DFixture()
|
| D | MaxPool2D.cpp | 30 "name": "OutputTensor", in MaxPool2DFixture()
|