Home
last modified time | relevance | path

Searched defs:node_output (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfunction_utils.h52 string node_output; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dcall.cc113 TfLiteTensor* node_output = context->tensors + node->outputs->data[i]; in ValidateAndResizeOutputs() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp2163 for (auto node_output : n->outputs()) { in ONNXShapeTypeInference() local
2208 for (auto node_output : n->outputs()) { in ONNXShapeTypeInference() local
2496 void UpdateShapeConstantIfReliable(torch::jit::Value* node_output) { in UpdateShapeConstantIfReliable()
H A Dpeephole.cpp606 for (auto node_output : n->outputs()) { in speculateOps() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dmatmul_builder.cc51 OpBuilder::TensorID* node_output) { in AddFullyConnectedHelper()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dutils.cc456 for (const auto& node_output : node_stat.output()) { in GetStatsStringFromRunMetadata() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dstep_stats_collector.cc163 NodeOutput* node_output = stats_->add_output(); in SetOutput() local
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dgraph.py754 def register_users_of(self, node_output): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpython_print.cpp596 [&](Value* block_input, Value* node_output) { in printLoop()
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dir_emitter.cpp4226 Value* node_output = in emitForkExpr() local
4262 Value* node_output = in emitAwaitableExpr() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp2186 Value* node_output = nullptr; in initJITBindings() local