Searched refs:OutputTensor (Results 1 – 7 of 7) sorted by relevance
| /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 | 287 struct OutputTensor { struct
291 OutputTensor(const Node* n, int i) : node(n), index(i) {} in OutputTensor() argument
292 OutputTensor() : node(nullptr), index(0) {} in OutputTensor() argument
581 OutputTensor cond_output,
582 std::vector<OutputTensor> body_inputs,
583 std::vector<OutputTensor> body_outputs,
|
| D | graph.cc | 687 OutputTensor cond_output, in AddWhileContext()
688 std::vector<OutputTensor> body_inputs, in AddWhileContext()
689 std::vector<OutputTensor> body_outputs, in AddWhileContext()
|
| /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 | 254 const std::vector<OutputTensor>& inputs, in GraphToFunctionDef()
255 const std::vector<OutputTensor>& outputs, in GraphToFunctionDef()
389 const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, in ProcessInputs()
426 std::vector<OutputTensor>* output_tensors) in ProcessOutputs()
498 std::vector<tensorflow::OutputTensor> input_tensors; in TF_GraphToFunction()
505 std::vector<tensorflow::OutputTensor> output_tensors; in TF_GraphToFunction()
|
| /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()
|