Home
last modified time | relevance | path

Searched defs:node_input (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
H A Dresolve_cluster.cc46 for (const std::string& node_input : node.input()) { in AddNodeToGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dgraph_transfer_info.proto35 repeated GraphTransferNodeInput node_input = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dcall.cc60 TfLiteTensor* node_input = context->tensors + node->inputs->data[i]; in ValidateAndResizeInputsIfNeeded() local
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_stats.cc210 string node_input = it->second->input(i); in AddGraph() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_observers.cpp1490 auto* node_input = n->input(i + input_offset); in insertObserversFor() local
1505 auto* node_input = n->input(i + input_offset); in insertObserversFor() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dconstant_propagation.cpp322 auto node_input = node->inputs().at(loop_input_offset + i); in removeExtraLoopOutputs() local
H A Dsymbolic_shape_runtime_fusion.cpp355 Value* node_input = guarded_node->inputs().at(i); in insertDynamicShapesGuard() local
H A Dsymbolic_shape_analysis.cpp997 auto node_input = node_inputs[i]; in joinPartialEvaluatedShapeGraphToLargeShapeGraph() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dgraph_transfer_info.pb.h2078 inline const ::tensorflow::GraphTransferNodeInput& GraphTransferNodeInputInfo::node_input(int index… in node_input() function
2091 GraphTransferNodeInputInfo::node_input() const { in node_input() function
/aosp_15_r20/external/ComputeLibrary/examples/
H A Dgraph_edsr.h685 INode *node_input = _graph.node(id_input); in setup() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dpeephole.cpp600 auto node_input = n->input()->node(); in speculateOps() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_constructor.cc923 const string& node_input = node_def->input(i); in AddControlDependencies() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dauto_mixed_precision.cc2012 GraphView::InputPort node_input(&node, 0); in NodeImplicitlyReadsNonResourceVariable() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc2615 std::vector<NodeInput> node_input = in TEST_P() local