Searched defs:OutputTensor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph.h | 405 struct OutputTensor { struct
409 OutputTensor(Node* n, int i) : node(n), index(i) {} in OutputTensor() argument
410 OutputTensor() : node(nullptr), index(0) {} in OutputTensor() argument
419 uint64 operator()(OutputTensor const& s) const; argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | import_model.cc | 1368 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody()
1549 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in Convert()
1604 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in ConvertFunctionArgAndRets()
2529 absl::InlinedVector<OutputTensor, 4>* arg_nodes, in InferMainFunctionType()
2662 mlir::MLIRContext* context, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsRetsAndTypesFromFunctionGraph()
2664 auto add_node = [](Node* node, absl::InlinedVector<OutputTensor, 4>* nodes) { in GetArgsRetsAndTypesFromFunctionGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unidirectional_sequence_gru.cc | 105 enum OutputTensor { enum
|
| H A D | lstm.cc | 2147 enum OutputTensor { enum
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_subgraphs_pass.cc | 888 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph()
|