Home
last modified time | relevance | path

Searched refs:node_output (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/executorch/backends/xnnpack/partition/config/
H A Dgeneric_node_configs.py58 node_output = list(node.users)[0]
60 node_output.op == "call_function"
61 and format_target_name(node_output.target.__name__) in self.fused_acts
63 quantized_deps.append(node_output)
64 fused_out_users = list(node_output.users.keys())
66 node_output = fused_out_users[0]
68 if not is_quant(node_output):
72 quantized_deps.append(node_output)
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfunction_utils.cc30 : node_name(node_name), node_output(output), position(position) { in FunctionDefTensorDesc()
31 full_str = strings::StrCat(node_name, ":", node_output, ":", position); in FunctionDefTensorDesc()
55 node_output = string(capture.data(), capture.size()); in FunctionDefTensorDesc()
H A Dfunction_utils_test.cc34 EXPECT_EQ(f.node_output, "y"); in TEST()
40 EXPECT_EQ(f2.node_output, ""); in TEST()
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
120 context, context->ResizeTensor(context, node_output, new_dims_array)); in ValidateAndResizeOutputs()
121 node_output->type = subgraph_output->type; in ValidateAndResizeOutputs()
/aosp_15_r20/external/pytorch/torch/utils/tensorboard/
H A D_pytorch_graph.py170 for node_output, outputSize in zip(node.outputs, node.outputstensor_size):
172 self.nodes_io[node_output] = NodeBase(
173 node_output,
/aosp_15_r20/external/executorch/devtools/inspector/tests/
H A Dinspector_utils_test.py242 node_output = ValueNode("output", [node_div])
252 node_output,
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dstep_stats_collector.cc163 NodeOutput* node_output = stats_->add_output(); in SetOutput() local
164 node_output->set_slot(slot); in SetOutput()
165 tensor->FillDescription(node_output->mutable_tensor_description()); in SetOutput()
/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
2166 ConstantValueMap::SetUseInferredType(node_output->debugName(), true); in ONNXShapeTypeInference()
2208 for (auto node_output : n->outputs()) { in ONNXShapeTypeInference() local
2209 UpdateShapeConstantIfReliable(node_output); in ONNXShapeTypeInference()
2496 void UpdateShapeConstantIfReliable(torch::jit::Value* node_output) { in UpdateShapeConstantIfReliable() argument
2497 if (ConstantValueMap::HasTypeReliable(node_output->debugName())) { in UpdateShapeConstantIfReliable()
2498 auto reliable = ConstantValueMap::GetTypeReliable(node_output->debugName()) in UpdateShapeConstantIfReliable()
2500 if (reliable && !ConstantValueMap::HasShape(node_output->debugName())) { in UpdateShapeConstantIfReliable()
2502 if (auto output_tensor_type = node_output->type()->cast<TensorType>()) { in UpdateShapeConstantIfReliable()
2505 UpdateShapeConstantValueMap(node_output, symbolic_sizes); in UpdateShapeConstantIfReliable()
H A Dpeephole.cpp606 for (auto node_output : n->outputs()) { in speculateOps() local
607 for (auto node_output_use : node_output->uses()) { in speculateOps()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dmatmul_builder.cc51 OpBuilder::TensorID* node_output) { in AddFullyConnectedHelper() argument
138 *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
459 const auto size = node_output.tensor_description() in GetStatsStringFromRunMetadata()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp2186 Value* node_output = nullptr; in initJITBindings() local
2203 node_output = in initJITBindings()
2211 jit::tracer::setValueTrace(retval, node_output); in initJITBindings()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpython_print.cpp596 [&](Value* block_input, Value* node_output) { in printLoop() argument
597 assignValue(block_input, node_output); in printLoop()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dgraph.py754 def register_users_of(self, node_output): argument
773 register(node_output)
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dir_emitter.cpp4226 Value* node_output = in emitForkExpr() local
4228 return std::make_shared<SimpleValue>(node_output); in emitForkExpr()
4262 Value* node_output = in emitAwaitableExpr() local
4264 return std::make_shared<SimpleValue>(node_output); in emitAwaitableExpr()