| /aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/ |
| H A D | xnnpack_graph_builder.cpp | 25 std::shared_ptr<torch::jit::Graph> graph, in optimizeAndTraceGraph()
38 std::shared_ptr<torch::jit::Graph>& graph, in buildXNNGraph()
106 void XNNGraph::checkOpsToDelegate(std::shared_ptr<torch::jit::Graph>& graph) { in checkOpsToDelegate()
137 for (auto val : _inputs) { in serializedXNNGraph() local
142 for (auto val : _outputs) { in serializedXNNGraph() local
153 for (auto val : _outputs) { in getGraphOutputShapes() local
162 void XNNGraph::defineAllNodes(std::shared_ptr<torch::jit::Graph>& graph) { in defineAllNodes()
207 for (auto val : _intermediate_tensors) { in defineAllTensorValues() local
281 void XNNGraph::gatherTensorValues(std::shared_ptr<torch::jit::Graph>& graph) { in gatherTensorValues()
316 bool XNNGraph::isGraphInput(torch::jit::Value* val) { in isGraphInput()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | freeze_module.cpp | 92 std::shared_ptr<Graph>& graph, in optimizeSubGraphs()
133 auto graph = toGraphFunction(*function).graph(); in run() local
144 auto graph = toGraphFunction(*function).graph(); in run() local
152 auto graph = toGraphFunction(*function).graph(); in run() local
212 bool _loadModulePath(Value* input, std::shared_ptr<Graph>& graph) { in _loadModulePath()
229 std::shared_ptr<Graph>& graph) { in getModulePath()
296 std::shared_ptr<Graph>& graph) { in findConstantAttr()
342 void recordMutableAttrs(std::shared_ptr<Graph>& graph) { in recordMutableAttrs()
409 for (auto& val : usedAttrs_) { in recordMutableAttrs() local
447 auto val = pair.value(); in overrideGradient() local
[all …]
|
| H A D | loop_unrolling.cpp | 20 bool isTrueConstant(Value* val) { in isTrueConstant()
59 auto graph = loop->owningGraph(); in inlineBody() local
96 Graph& graph, in insertBlockCopy()
122 auto graph = body->owningGraph(); in repeatBody() local
150 Graph* graph = loop->owningGraph(); in replaceLoopCounter() local
167 Graph* graph = loop->owningGraph(); in unroll() local
258 bool LoopsPeeler::run(const std::shared_ptr<Graph>& graph) { in run()
303 bool PeelProfilingLoops(const std::shared_ptr<Graph>& graph) { in PeelProfilingLoops()
321 auto graph = n->owningGraph(); in PeelLoop() local
375 bool UnrollLoops(std::shared_ptr<Graph>& graph) { in UnrollLoops()
[all …]
|
| H A D | canonicalize_graph_fuser_ops.cpp | 10 Value* val; member
56 auto graph = it->owningGraph(); in CanonicalizeOps() local
75 auto* graph = it->owningGraph(); in CanonicalizeOps() local
92 void CanonicalizeOps(const std::shared_ptr<Graph>& graph) { in CanonicalizeOps()
|
| H A D | tensorexpr_fuser.cpp | 62 Graph* graph = sizes[0]->owningGraph(); in broadcastSizes() local
154 void setTensorExprFuserEnabled(bool val) { in setTensorExprFuserEnabled()
173 void setTensorExprDynamicShapeFusionEnabled(bool val) { in setTensorExprDynamicShapeFusionEnabled()
251 void RemoveProfileNodesAndSpecializeTypes(std::shared_ptr<Graph>& graph) { in RemoveProfileNodesAndSpecializeTypes()
295 void RemoveTensorTypeSpecializations(std::shared_ptr<Graph>& graph) { in RemoveTensorTypeSpecializations()
415 std::shared_ptr<Graph> graph, in TensorExprFuser()
434 Graph* graph = fusion_group->owningGraph(); in buildShapeExpressions() local
1326 std::shared_ptr<Graph>& graph, in FuseTensorExprs()
|
| H A D | remove_dropout.cpp | 14 const IValue& val = optional_ivalue.value(); in isDropoutRemovable() local
48 void removeDropout(std::shared_ptr<Graph>& graph) { in removeDropout()
56 auto graph = module.get_method("forward").graph(); in removeDropout() local
|
| H A D | peephole_dict_idioms.cpp | 60 auto val = ivalue_converter_(ivalue); in get() local
144 explicit PeepholeOptimizeDictIdiomsImpl(std::shared_ptr<Graph> graph) in PeepholeOptimizeDictIdiomsImpl()
265 bool PeepholeOptimizeDictIdioms(const std::shared_ptr<Graph>& graph) { in PeepholeOptimizeDictIdioms()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | UnaryOp.cpp | 24 ComputeGraph* graph, in resize_unary_op_node()
35 ComputeGraph& graph, in add_unary_op_node()
70 float get_val_or_inf(ComputeGraph& graph, const ValueRef& val, bool max) { in get_val_or_inf()
117 void gelu(ComputeGraph& graph, const std::vector<ValueRef>& args) { in gelu()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 51 Status RunGraph(const Graph& graph, in RunGraph()
67 Graph graph(OpRegistry::Global()); in TEST() local
94 Graph graph(OpRegistry::Global()); in TEST() local
135 const int num_outputs, Graph* graph, in CreateSplitTensorGraph()
162 const int num_outputs, Graph* graph, in CreateSplitResourceGraph()
213 Graph graph(OpRegistry::Global()); in TEST_P() local
231 Graph graph(OpRegistry::Global()); in TEST_P() local
249 Graph graph(OpRegistry::Global()); in TEST_P() local
267 Graph graph(OpRegistry::Global()); in TEST_P() local
284 Graph graph(OpRegistry::Global()); in TEST_P() local
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | profiling_graph_executor_impl.cpp | 221 auto set_requires_grad = [](const TensorTypePtr& t, Value* val) -> bool { in setRequiresGradOnDiffGraph()
329 void runNooptPassPipeline(std::shared_ptr<Graph>& graph) { in runNooptPassPipeline()
344 static void runPreAutodiffPassPipeline(std::shared_ptr<Graph>& graph) { in runPreAutodiffPassPipeline()
421 std::shared_ptr<Graph>& graph, in runNoGradOptimizations()
538 std::shared_ptr<Graph>& graph) { in runProfilingInsensitiveOptimizations()
599 const std::shared_ptr<Graph>& graph, in ProfilingGraphExecutorImpl()
732 Graph* graph, in insertFallbackFunctionCall()
756 auto graph = std::make_shared<Graph>(); in createFallbackPathFunction() local
803 std::shared_ptr<Graph>& graph) { in runFinalOptimizations()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | grpc_session_debug_test.cc | 53 Graph graph(OpRegistry::Global()); in CreateGraphDef() local
74 void IsSingleFloatValue(const Tensor& val, float expected_val) { in IsSingleFloatValue()
152 GraphDef graph; in TEST_F() local
215 void SetDevice(GraphDef* graph, const string& name, const string& dev) { in SetDevice()
231 Graph graph(OpRegistry::Global()); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | cast_all_constant_to_floating.cpp | 18 auto graph = block->owningGraph(); in CastAllConstantToFloating() local
28 auto val = node->t(attr::value); in CastAllConstantToFloating() local
69 void CastAllConstantToFloating(const std::shared_ptr<Graph>& graph) { in CastAllConstantToFloating()
|
| H A D | fixup_onnx_controlflow.cpp | 17 Node* CreateCastToBoolNode(Value* val, Graph* graph) { in CreateCastToBoolNode()
27 Graph* graph, in InsertCastForCond()
295 auto* graph = node->owningGraph(); in FixupONNXLoopNodeInputs() local
384 Graph* graph, in InferShapeTypeForUninitializedOutput()
466 auto* graph = if_node->owningGraph(); in ONNXFixupUninitializedOutput() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | impl.cpp | 59 std::string iValueToString(const c10::IValue& val) { in iValueToString()
130 bool canEnableStaticRuntime(const std::shared_ptr<torch::jit::Graph>& graph) { in canEnableStaticRuntime()
147 for (const auto* val : value_set) { in dumpValueSet() local
157 std::shared_ptr<torch::jit::Graph>& graph, in OptimizeGraph()
224 bool IsSelfInGraphInput(std::shared_ptr<torch::jit::Graph>& graph) { in IsSelfInGraphInput()
229 bool removeSelfFromGraphInput(std::shared_ptr<torch::jit::Graph>& graph) { in removeSelfFromGraphInput()
269 std::shared_ptr<torch::jit::Graph> graph, in PrepareGraphForStaticModule()
309 auto graph = module.get_method("forward").graph(); in PrepareForStaticModule() local
320 const std::shared_ptr<torch::jit::Graph>& graph, in PrepareForStaticModule()
1859 const Value* val = pnode.node()->output(i); in check_for_memory_leak() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_cond.cc | 134 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue()
159 StateMap::StateMap(Graph* graph) { in StateMap()
289 FunctionalizeCond::FunctionalizeCond(Graph* graph, in FunctionalizeCond()
499 Graph* graph) { in AddSwitchNodeAlongEdge()
527 Status Conditional::ExtractBodies(Graph* graph) { in ExtractBodies()
747 Status Conditional::BuildIfNode(Graph* graph, in BuildIfNode()
841 Graph* graph, in AddInputEdges()
878 Graph* graph, in AddOutputEdges()
917 Graph* graph, FunctionLibraryDefinition* library, in BuildAndReplace()
1265 OutputTensor val; in RemoveRedundantSwitch() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/multibox_detector/ |
| H A D | main.cc | 127 tensorflow::GraphDef graph; in ReadTensorFromImageFile() local
154 tensorflow::GraphDef graph; in SaveImage() local
196 tensorflow::GraphDef graph; in GetTopDetections() local
251 uint8 val = i == 2 ? 255 : 0; in DrawBox() local
338 string graph = in main() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | utils_test.cc | 213 GraphDef graph; in TEST_F() local
240 GraphDef graph; in TEST_F() local
353 GraphDef graph; in TEST_F() local
601 void TestSetTensorValue(DataType type, int val, bool success, in TestSetTensorValue()
697 GraphDef graph = test::CreateRandomGraph(size); in BM_NodeMapConstruct() local
707 GraphDef graph = test::CreateRandomGraph(size); in BM_ImmutableNodeMapConstruct() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | executor_test.cc | 76 void Create(std::unique_ptr<const Graph> graph) { in Create()
113 Tensor V(const float val) { in V()
120 Tensor VI(const int32_t val) { in VI()
127 Tensor VB(const bool val) { in VB()
134 Tensor VD(const double val) { in VD()
533 Tensor val(DT_FLOAT, TensorShape({})); in BM_FeedInputFetchOutput() local
596 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); in BM_WhileLoopHelper() local
|
| H A D | direct_session.cc | 409 Status DirectSession::Create(const GraphDef& graph) { in Create()
413 Status DirectSession::Create(GraphDef&& graph) { in Create()
426 Status DirectSession::Extend(const GraphDef& graph) { in Extend()
430 Status DirectSession::Extend(GraphDef&& graph) { in Extend()
436 Status DirectSession::ExtendLocked(GraphDef&& graph) { in ExtendLocked()
496 const DebugOptions& debug_options, Graph* graph, Device* device) { in DecorateAndPublishGraphForDebug()
799 const Graph* graph = partition.graph.get(); in RunInternal() local
1251 const Graph* graph = executors_and_keys->graph.get(); in CheckFetch() local
1735 std::unique_ptr<Graph>* graph = &partition.second; in CreateGraphs() local
1895 Status GetArg(int index, const Tensor** val) override { in GetArg()
[all …]
|
| H A D | single_threaded_executor.cc | 85 Status Initialize(const Graph& graph) { in Initialize()
455 TensorValue val = ctx.release_output(j); in Run() local
580 Status NewExecutor(const LocalExecutorParams& params, const Graph& graph, in NewExecutor()
594 const Graph& graph, Executor** executor) { in NewSingleThreadedExecutor()
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/utils/ |
| H A D | test_utils.cpp | 393 void fill_vtensor(api::vTensor& vten, float val, bool iota) { in fill_vtensor()
433 ComputeGraph& graph, in fill_vtensor()
435 float val, in fill_vtensor()
514 ComputeGraph& graph, in execute_graph_and_check_output()
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | vulkan_compute_api_test.cpp | 1146 ComputeGraph graph(config); in TEST() local
1159 ComputeGraph graph(config); in TEST() local
1171 ComputeGraph graph(config); in TEST() local
1187 ComputeGraph graph(config); in TEST() local
1202 ComputeGraph graph(config); in TEST() local
1212 ComputeGraph graph(config); in TEST() local
1257 ComputeGraph graph(config); in TEST() local
1280 float val = -i + 2.0f; in TEST() local
1306 ComputeGraph graph(config); in TEST() local
1355 ComputeGraph graph(config); in TEST() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_session_test.cc | 45 Graph graph(OpRegistry::Global()); in CreateGraphDef() local
65 static void IsSingleFloatValue(const Tensor& val, float expected_val) { in IsSingleFloatValue()
88 GraphDef graph; in TEST() local
124 GraphDef graph; in TEST() local
169 GraphDef graph; in TEST() local
200 GraphDef graph; in TEST() local
232 GraphDef graph; in TEST() local
263 GraphDef graph; in TEST() local
290 GraphDef graph; in TEST() local
329 Graph graph(OpRegistry::Global()); in FindMaxEigen() local
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/ |
| H A D | compiler.cpp | 112 const Value* val = queue.back(); in getInputDependencies() local
179 auto graph = normalizeGraphForCache(fusion_group->g(attr::Subgraph)); in registerFusion() local
207 auto graph = spec.graph()->copy(); in compileKernel() local
|
| /aosp_15_r20/art/compiler/optimizing/ |
| H A D | optimizing_unit_test.h | 110 inline void RemoveSuspendChecks(HGraph* graph) { in RemoveSuspendChecks()
146 HGraph* graph, in AdjacencyListGraph()
264 HGraph* graph = CreateGraph(&handles); variable
906 HParameterValue* val = new (GetAllocator()) HParameterValue( variable
913 bool CheckGraph(HGraph* graph, std::ostream& oss) { in CheckGraph()
1016 explicit PatternMatchGraphVisitor(HGraph* graph, Inst... handlers) : HGraphVisitor(graph) { in PatternMatchGraphVisitor()
|