Searched defs:OutputTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 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
409 uint64 operator()(OutputTensor const& s) const; argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1316 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody()
1496 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in Convert()
1551 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in ConvertFunctionArgAndRets()
2460 absl::InlinedVector<OutputTensor, 4>* arg_nodes, in InferMainFunctionType()
2570 mlir::MLIRContext* context, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsRetsAndTypesFromFunctionGraph()
2572 auto add_node = [](Node* node, absl::InlinedVector<OutputTensor, 4>* nodes) { in GetArgsRetsAndTypesFromFunctionGraph()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_gru.cc | 105 enum OutputTensor { enum
|
| D | lstm.cc | 2143 enum OutputTensor { enum
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 911 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph()
|