Home
last modified time | relevance | path

Searched refs:OutputTensor (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/graph/
Dwhile_context.h38 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_;
Dwhile_context.cc23 OutputTensor cond_output, in WhileContext()
24 std::vector<OutputTensor> body_inputs, in WhileContext()
25 std::vector<OutputTensor> body_outputs) in WhileContext()
Dgraph.h62 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() argument
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,
Dgraph.cc307 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()
Dnode_builder.h53 NodeOut(OutputTensor t);
Dnode_builder.cc32 NodeBuilder::NodeOut::NodeOut(OutputTensor t) : NodeOut(t.node, t.index) {} in NodeOut()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.h61 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_;
Dfunctionalize_cond.cc66 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 …]
Dfunctionalize_cond_test.cc79 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/
Dclone_constants_for_better_clustering_test.cc81 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()
Dencapsulate_subgraphs_pass.cc130 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 …]
Dencapsulate_subgraphs_pass.h43 const std::vector<OutputTensor>& arg_source_tensors,
Dextract_outside_compilation_pass.h54 Status operator()(const std::vector<OutputTensor>&,
Dextract_outside_compilation_pass_test.cc61 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()
Dencapsulate_xla_computations_pass.cc113 Status RewriteSubgraph(const std::vector<OutputTensor>& arg_source_tensors, in RewriteSubgraph()
Dencapsulate_subgraphs_pass_test.cc802 [&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/
Dwhile_loop.cc30 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/
Dc_api_function.cc302 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()
Dc_api.cc87 using tensorflow::OutputTensor;
2805 OutputTensor tensor(&output.oper->node, output.index); in TF_TryEvaluateConstant()
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients.cc31 Output ToOutput(OutputTensor output_tensor) { in ToOutput()
36 const std::vector<OutputTensor>& output_tensors) { in ToOutputVector()
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.h57 OutputTensor tensor, const ShapeRefiner& refiner,
Dlower_if_op.cc79 OutputTensor pred_;
Deval_const_tensor.cc308 Status EvaluateConstantTensor(OutputTensor tensor, const ShapeRefiner& refiner, in EvaluateConstantTensor()
Dshape_refiner.cc422 OutputTensor tensor(input_edge->src(), input_edge->src_output()); in EvaluateConstantTensorForEdge()
/external/tensorflow/tensorflow/lite/kernels/
Dlstm.cc654 enum OutputTensor { enum

12