Searched defs:OutputTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 385 struct OutputTensor { struct
389 OutputTensor(Node* n, int i) : node(n), index(i) {} in OutputTensor() argument
390 OutputTensor() : node(nullptr), index(0) {} in OutputTensor() function
399 uint64 operator()(OutputTensor const& s) const; argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1215 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody()
1425 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in Convert()
1480 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in ConvertFunctionArgAndRets()
2315 absl::InlinedVector<OutputTensor, 4>* arg_nodes, in InferMainFunctionType()
2425 mlir::MLIRContext* context, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsRetsAndTypesFromFunctionGraph()
2427 auto add_node = [](Node* node, absl::InlinedVector<OutputTensor, 4>* nodes) { in GetArgsRetsAndTypesFromFunctionGraph()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 97 enum OutputTensor { enum
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 914 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 2174 enum OutputTensor { enum
|