| /aosp_15_r20/external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| H A D | constraint.py | 217 def __init__(self, tensor_size, input_var, dim_replace, index, output): argument
258 def __init__(self, tensor_size, input_var, index1, index2, output): argument
299 def __init__(self, tensor_size, index, res, input_var): argument
328 def __init__(self, tensor_size, index_tuple, res, input_var): argument
359 …def __init__(self, conv_result, input_var, c_out, kernel, padding, stride, dilation, matching_cons… argument
394 …def __init__(self, maxpool_result, input_var, kernel, padding, stride, dilation, matching_constrai… argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | numpy_dataset.py | 28 def init_var_from_numpy(input_var, numpy_input, session): argument
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/batchnormalization/impl/SVE/ |
| H A D | fp32.cpp | 52 … const auto input_var = reinterpret_cast<const float *>(var->ptr_to_element(Coordinates(0, 0))); in fp32_sve_batch_normalization() local
|
| H A D | fp16.cpp | 52 …const auto input_var = reinterpret_cast<const float16_t *>(var->ptr_to_element(Coordinates(0, 0)… in fp16_sve_batch_normalization() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/generic/ |
| H A D | impl.cpp | 56 …const auto input_var = reinterpret_cast<const ScalarType *>(bn_var->ptr_to_element(Coordinates(0… in fused_batch_normalization_conv() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nhwc/neon/ |
| H A D | impl.cpp | 56 …const auto input_var = reinterpret_cast<const ScalarType *>(bn_var->ptr_to_element(Coordinates(0… in fused_batch_normalization_dwc_nhwc() local
|
| /aosp_15_r20/external/mesa3d/src/compiler/glsl/ |
| H A D | gl_nir_link_varyings.c | 3258 nir_variable *input_var; in get_matching_input() local
3686 should_add_varying_match_record(nir_variable *const input_var, in should_add_varying_match_record()
3807 nir_variable *const input_var = in assign_initial_varying_locations() local
3903 nir_variable *const input_var = in assign_initial_varying_locations() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/batchnormalization/impl/NEON/ |
| H A D | fp16.cpp | 61 …const auto input_var = reinterpret_cast<const float16_t *>(var->ptr_to_element(Coordinates(0, 0)… in batch_normalization() local
|
| H A D | fp32.cpp | 60 … const auto input_var = reinterpret_cast<const float *>(var->ptr_to_element(Coordinates(0, 0))); in batch_normalization() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/nchw/ |
| H A D | all.cpp | 57 …const auto input_var = reinterpret_cast<const ScalarType *>(bn_var->ptr_to_element(Coordinates(0… in fused_batch_normalization_dwc_nchw() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NEBatchNormalizationLayerKernel.cpp | 173 const auto input_var = reinterpret_cast<const T *>(_var->ptr_to_element(Coordinates(0, 0))); in batch_normalization_nchw() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_utils.py | 124 def forward(self, input_var): argument
|
| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_linking_helpers.c | 1082 replace_duplicate_input(nir_shader *shader, nir_variable *input_var, in replace_duplicate_input()
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 395 self, model, input_var, target, loss, run_checkpoint=False, use_reentrant=True argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 4204 self, model, input_var, target, loss, scale_factor=1.0, memory_format=None argument
|
| /aosp_15_r20/external/deqp/external/openglcts/modules/gl/ |
| H A D | gl4cEnhancedLayoutsTests.cpp | 13135 static const GLchar *input_var = "in Data dataARRAY;\n"; in test() local
|