| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | graph_rewrite_helper.cpp | 24 const std::unordered_map<std::string, Value*>& vmap) { in getValue()
31 const std::unordered_map<std::string, Value*>& vmap) { in getIValue()
37 const std::unordered_map<std::string, Value*>& vmap) { in getConvParams()
156 const std::unordered_map<std::string, Value*>& vmap) { in replaceConvolutionWithAtenConv()
167 const std::unordered_map<std::string, Value*>& vmap) { in replaceConvolutionWithAtenConv()
178 const std::unordered_map<std::string, Value*>& vmap) { in replaceConvolutionWithAtenConv()
190 const std::unordered_map<std::string, Value*>& vmap) { in replaceConvolutionWithAtenConv()
202 const std::unordered_map<std::string, Value*>& vmap) { in replaceConvolutionWithAtenConv()
214 const std::unordered_map<std::string, Value*>& vmap) { in replaceConvolutionWithAtenConv()
267 const std::unordered_map<std::string, Value*>& vmap) { in isClampFusable()
|
| H A D | fuse_linear.cpp | 20 const std::unordered_map<std::string, Value*>& vmap) { in FuseLinear()
28 const std::unordered_map<std::string, Value*>& vmap) { in FuseLinear()
123 const std::unordered_map<std::string, Value*>& vmap) { in SwapFunctionalLinear()
|
| H A D | vulkan_rewrite.cpp | 309 const std::unordered_map<std::string, Value*>& vmap) { in insertPrePackedGruOp()
333 const std::unordered_map<std::string, Value*>& vmap) { in insertPrePackedLstmOp()
|
| H A D | graph_rewrite_helper.h | 34 std::unordered_map<std::string, Value*> vmap; member
|
| H A D | frozen_conv_add_relu_fusion_cuda.cpp | 80 const std::unordered_map<std::string, Value*>& vmap) { in fuseFrozenConvAddReluImpl()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | helper.cpp | 665 const std::unordered_map<std::string, Value*>& vmap, in is_int_constant()
675 const std::unordered_map<std::string, Value*>& vmap, in is_functional()
699 const std::unordered_map<std::string, Value*>& vmap, in is_module()
713 const std::unordered_map<std::string, Value*>& vmap) { in aten_add_alpha_is_one()
719 const std::unordered_map<std::string, Value*>& vmap) { in is_functional_relu()
725 const std::unordered_map<std::string, Value*>& vmap) { in is_relu_module()
732 const std::unordered_map<std::string, Value*>& vmap) { in is_linear_module()
739 const std::unordered_map<std::string, Value*>& vmap) { in is_conv1d_module()
746 const std::unordered_map<std::string, Value*>& vmap) { in is_conv2d_module()
753 const std::unordered_map<std::string, Value*>& vmap) { in is_conv3d_module()
[all …]
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | test_subgraph_rewriter.cpp | 29 std::unordered_map<std::string, Value*> vmap; in TEST() local
34 const std::unordered_map<std::string, Value*>& vmap) { in TEST()
41 const std::unordered_map<std::string, Value*>& vmap) { in TEST()
48 const std::unordered_map<std::string, Value*>& vmap) { in TEST()
100 std::unordered_map<std::string, Value*> vmap; in TEST() local
105 const std::unordered_map<std::string, Value*>& vmap) { in TEST()
254 std::unordered_map<std::string, Value*> vmap; in TEST() local
259 const std::unordered_map<std::string, Value*>& vmap) { in TEST()
|
| H A D | test_alias_analysis.cpp | 436 std::unordered_map<std::string, Value*> vmap; in TEST() local
454 std::unordered_map<std::string, Value*> vmap; in TEST() local
684 std::unordered_map<std::string, Value*> vmap; in TEST() local
702 std::unordered_map<std::string, Value*> vmap; in TEST() local
739 std::unordered_map<std::string, Value*> vmap; in TEST() local
938 std::unordered_map<std::string, Value*> vmap; in TEST() local
966 std::unordered_map<std::string, Value*> vmap; in TEST() local
1009 std::unordered_map<std::string, Value*> vmap; in TEST() local
1041 std::unordered_map<std::string, Value*> vmap; in TEST() local
1081 std::unordered_map<std::string, Value*> vmap; in TEST() local
[all …]
|
| H A D | test_irparser.cpp | 44 std::unordered_map<std::string, Value*> vmap; in TEST() local
269 std::unordered_map<std::string, Value*> vmap; local
303 std::unordered_map<std::string, Value*> vmap; local
|
| H A D | test_interpreter.cpp | 25 std::unordered_map<std::string, Value*> vmap; in makeInterp() local
228 std::unordered_map<std::string, Value*> vmap; in TEST() local
|
| H A D | test_misc.cpp | 2048 std::unordered_map<std::string, Value*> vmap; in TEST() local
2910 std::unordered_map<std::string, Value*> vmap; in TEST() local
2930 std::unordered_map<std::string, Value*> vmap; in TEST() local
2947 std::unordered_map<std::string, Value*> vmap; in TEST() local
2974 std::unordered_map<std::string, Value*> vmap; in TEST() local
3142 std::unordered_map<std::string, Value*> vmap; in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | autograd_function_db.py | 47 def vmap(info, in_dims, input): member in NumpyCube
127 def vmap(info, in_dims, x, y): member in NumpyMul
198 def vmap(info, in_dims, x): member in NumpyExp_
238 def vmap(info, in_dims, x, dim): member in NumpySort
316 def vmap(info, in_dims, x, ind, ind_inv, dim): member in NumpyTake
384 def vmap(info, in_dims, x, idx): member in Select
495 def vmap(info, in_dims, x, idx): member in ForwardHasDefaultArgs
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | irparser.cpp | 33 std::unordered_map<std::string, Value*>& vmap, in IRParser()
78 std::unordered_map<std::string, Value*>& vmap; member in torch::jit::IRParser
111 std::unordered_map<std::string, Value*>& vmap, in parseIR()
121 std::unordered_map<std::string, Value*> vmap; in parseIR() local
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_module.cc | 1121 std::unordered_map<std::string, Value*> vmap; in TEST() local
1160 std::unordered_map<std::string, Value*> vmap; in TEST() local
1208 std::unordered_map<std::string, Value*> vmap; in TEST() local
1244 std::unordered_map<std::string, Value*> vmap; in TEST() local
1266 std::unordered_map<std::string, Value*> vmap; in TEST() local
1343 std::unordered_map<std::string, Value*> vmap; in testAssignStorageToManagedTensors() local
|
| H A D | test_utils.cc | 95 std::unordered_map<std::string, Value*> vmap; in GraphStaticRuntimeContext() local
256 std::unordered_map<std::string, Value*> vmap; in getGraphFromIR() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | passes.cpp | 1156 const std::unordered_map<std::string, Value*>& vmap) { in eliminatePermuteOpsSumPattern()
1202 const std::unordered_map<std::string, Value*>& vmap) { in eliminatePermuteOpsSoftmaxPattern()
1384 const std::unordered_map<std::string, Value*>& vmap) { in UseInPlaceGetRealInputsFromOptionalInputsV2()
1420 const std::unordered_map<std::string, Value*>& vmap) { in FuseClampNaNToNum()
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 1385 def vmap(infos, in_dims, x): member in TestAutogradFunctionVmapAPI.test_has_vmap_staticmethod_and_has_generate_vmap_rule.NumpyCube
1409 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_info_object.Id
1434 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_in_dims_single_input.Id
1458 def vmap(info, in_dims, x, y): member in TestAutogradFunctionVmapAPI.test_in_dims_multiple_inputs.Id
1482 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_skips_empty_layer.Id
1504 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_none_returns.Zeros
1524 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_none_returns.TwoZeros
1550 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_should_have_two_returns.Zeros
1570 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_should_have_two_returns.TwoZeros
1591 def vmap(info, in_dims, input): member in TestAutogradFunctionVmapAPI.test_incompatible_out_dims_error_msg.Zeros
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | unpack_quantized_weights.cpp | 265 std::unordered_map<std::string, Value*> vmap; in unpackQuantizedWeightsHelper() local
719 std::unordered_map<std::string, Value*> vmap; in insertPermutesHelper() local
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/llvm/IR/ |
| D | ValueSymbolTable.h | 136 ValueMap vmap; ///< The map that holds the symbol table. variable
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/IR/ |
| H A D | ValueSymbolTable.h | 135 ValueMap vmap; ///< The map that holds the symbol table. variable
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/llvm/IR/ |
| D | ValueSymbolTable.h | 136 ValueMap vmap; ///< The map that holds the symbol table. variable
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/llvm/IR/ |
| D | ValueSymbolTable.h | 136 ValueMap vmap; ///< The map that holds the symbol table. variable
|
| /aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/llvm/IR/ |
| D | ValueSymbolTable.h | 136 ValueMap vmap; ///< The map that holds the symbol table. variable
|
| /aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/IR/ |
| H A D | ValueSymbolTable.h | 130 ValueMap vmap; ///< The map that holds the symbol table. variable
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | apis.py | 32 def vmap( function
|