| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorShape.cu | 52 __device__ inline void stream_load128(uint4& val, const T* addr) { in stream_load128()
67 __device__ inline void stream_store128(T* addr, const uint4& val) { in stream_store128()
85 static __device__ inline void load128(uint4& val, const char* addr) { in load128()
92 __device__ inline void load128<uint4>(uint4& val, const char* addr) { in load128()
96 static __device__ inline void load128(uint4& val, const char* addr) { in load128()
140 uint4 val; in copy_chunk() local
189 int64_t dim) { in get_split_base_addrs()
216 int64_t dim) { in get_split_chunk_sizes()
228 static inline int64_t get_chunk_stride(const at::Tensor& tensor, int64_t dim) { in get_chunk_stride()
237 static inline int64_t get_num_chunks(const at::Tensor& tensor, int64_t dim) { in get_num_chunks()
[all …]
|
| H A D | ReduceOps.cpp | 30 void norm_kernel_cuda(TensorIterator& iter, const Scalar& val) { in norm_kernel_cuda()
52 void min_kernel_impl(const Tensor& result, const Tensor& indice, const Tensor& self, int64_t dim, b… in min_kernel_impl()
57 void max_kernel_impl(const Tensor& result, const Tensor& indice, const Tensor& self, int64_t dim, b… in max_kernel_impl()
63 const Tensor& self, int64_t dim, bool keepdim, Tensor& min_result, Tensor& max_result) { in aminmax_kernel_impl()
|
| H A D | ScanUtils.cuh | 166 const auto val = c10::load(self); in tensor_kernel_scan_outer_dim_with_indices() local
181 inline void check_fits_in_unsigned(int64_t val, const char* name) { in check_fits_in_unsigned()
191 int dim, scalar_t init, BinaryFunction binary_op) { in scan_outer_dim_with_indices()
242 …dices(const TensorBase& self, const TensorBase& values, const TensorBase& indices, //int64_t dim) { in scan_dim_with_indices()
392 int dim, scalar_t init, BinaryFunction binary_op) { in scan_outer_dim()
445 int64_t dim, scalar_t init, BinaryFunction binary_op) { in scan_dim()
|
| H A D | Sorting.cu | 123 scalar_t val = doLdg(&inputSliceStart[i * inputWithinSliceStride]); in gatherMedian() local
162 scalar_t val = doLdg(&inputSliceStart[i * inputWithinSliceStride]); in gatherMedian() local
249 const TensorBase &self, int64_t dim, int64_t k) { in launch_kthvalue_kernel()
265 const TensorBase &self, int64_t dim, bool ignore_nan) { in launch_median_kernel()
|
| /aosp_15_r20/frameworks/rs/tests/java_api/RSUnitTests/src/com/android/rs/unittest/ |
| H A D | UT_math_agree.java | 48 private Float2 pack_f2(float[] val) { in pack_f2()
53 private Float3 pack_f3(float[] val) { in pack_f3()
58 private Float4 pack_f4(float[] val) { in pack_f4()
63 private Byte2 pack_b2(byte[] val) { in pack_b2()
68 private Byte3 pack_b3(byte[] val) { in pack_b3()
73 private Byte4 pack_b4(byte[] val) { in pack_b4()
78 private Short2 pack_s2(short[] val) { in pack_s2()
83 private Short3 pack_s3(short[] val) { in pack_s3()
88 private Short4 pack_s4(short[] val) { in pack_s4()
93 private Int2 pack_i2(int[] val) { in pack_i2()
[all …]
|
| /aosp_15_r20/frameworks/rs/tests/java_api/RSUnitTests/supportlibsrc_gen/com/android/rs/unittest/ |
| H A D | UT_math_agree.java | 50 private Float2 pack_f2(float[] val) { in pack_f2()
55 private Float3 pack_f3(float[] val) { in pack_f3()
60 private Float4 pack_f4(float[] val) { in pack_f4()
65 private Byte2 pack_b2(byte[] val) { in pack_b2()
70 private Byte3 pack_b3(byte[] val) { in pack_b3()
75 private Byte4 pack_b4(byte[] val) { in pack_b4()
80 private Short2 pack_s2(short[] val) { in pack_s2()
85 private Short3 pack_s3(short[] val) { in pack_s3()
90 private Short4 pack_s4(short[] val) { in pack_s4()
95 private Int2 pack_i2(int[] val) { in pack_i2()
[all …]
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_scatter.cpp | 29 int64_t dim, in scatter_src_helper()
69 int64_t dim, in scatter_value_helper()
71 CTYPE_VAL val, in scatter_value_helper()
108 int64_t dim, in scatter_src_out()
138 int64_t dim, in scatter_value_out()
158 CTYPE_VAL val; in scatter_value_out() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 70 inline int64_t GetDim(absl::Span<const int64_t> data, DimIndex dim) { in GetDim()
74 inline void SetDim(absl::Span<int64_t> data, DimIndex dim, int64_t value) { in SetDim()
78 inline void SetDim(std::vector<int64_t>* data, DimIndex dim, int64_t value) { in SetDim()
262 int64_t spatial_dim(DimIndex dim) const { in spatial_dim()
307 BatchDescriptor& set_spatial_dim(DimIndex dim, int64_t value) { in set_spatial_dim()
431 FilterDescriptor& set_spatial_dim(DimIndex dim, int64_t value) { in set_spatial_dim()
459 int64_t input_filter_dim(DimIndex dim) const { in input_filter_dim()
559 ConvolutionDescriptor& set_zero_padding(DimIndex dim, int64_t value) { in set_zero_padding()
571 ConvolutionDescriptor& set_filter_stride(DimIndex dim, int64_t value) { in set_filter_stride()
583 ConvolutionDescriptor& set_dilation_rate(DimIndex dim, int64_t value) { in set_dilation_rate()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels_test.cc | 344 std::unique_ptr<char[]> val(new char[5]); in TEST_F() local
423 TF_Tensor* val; in TEST_F() local
542 int64_t val; in TEST_F() local
592 float val; in TEST_F() local
642 unsigned char val; in TEST_F() local
692 TF_DataType val; in TEST_F() local
1054 int64_t dim = 1; in TEST_F() local
1082 int64_t dim = 0; in TEST_F() local
1106 int64_t dim[2] = {2, 3}; in TEST_F() local
1135 int64_t dim = 1; in TEST_F() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorAdvancedIndexing.cpp | 191 int64_t dim, in scatter_meta_impl()
326 int64_t dim, in index_func_meta_impl()
558 for (const auto dim : c10::irange(indices_list.size())) { in AdvancedIndex() local
988 …check_sizes = [&ep_sizes, &ep_strides, &numel](IntArrayRef a, IntArrayRef b, int64_t dim) -> bool { in TORCH_IMPL_FUNC()
1063 int64_t dim, in index_reduce_func_impl()
1179 scalar_t val; in index_reduce_func_impl() local
1301 Tensor & index_select_out_cpu_(const Tensor & self, int64_t dim, const Tensor & index, Tensor & res… in index_select_out_cpu_()
1463 Tensor index_select_cpu_(const Tensor & self, int64_t dim, const Tensor & index) { in index_select_cpu_()
1468 Tensor index_select_quantized_cpu_(const Tensor & self, int64_t dim, const Tensor & index) { in index_select_quantized_cpu_()
1475 Tensor index_select_backward_symint(const Tensor& grad, c10::SymIntArrayRef self_sizes, int64_t dim… in index_select_backward_symint()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 170 def linalg_cross(self, other, *, dim=-1): argument
200 def cummaxmin(self, dim): argument
211 def logcumsumexp(self, dim): argument
218 def _exec_fft(out, self, out_sizes, dim, forward): argument
274 def meta_fft_c2c(self, dim, normalization, forward): argument
293 def meta_fft_r2c(self, dim, normalization, onesided): argument
369 def meta_fft_c2r(self, dim, normalization, lastdim): argument
401 def inferUnsqueezeGeometry(tensor, dim): argument
411 def meta_unsqueeze_(self, dim): argument
578 def meta_index_select(self, dim, index): argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/ |
| H A D | util.cpp | 296 std::string ivalueToStr(const c10::IValue& val, bool isString) { in ivalueToStr()
321 for (const auto& val : list) { in ivalueListToStr() local
685 for (int64_t dim : mat1_size) { in computeFlops() local
689 for (int64_t dim : mat2_size) { in computeFlops() local
731 for (int64_t dim : mat1_size) { in computeFlops() local
736 for (int64_t dim : mat2_size) { in computeFlops() local
756 for (int64_t dim : mat_size) { in computeFlops() local
775 for (int64_t dim : mat_size) { in computeFlops() local
|
| /aosp_15_r20/external/freetype/src/pshinter/ |
| H A D | pshalgo.c | 145 FT_Int mask = 0, val = 0; in psh_hint_table_record_mask() local
246 FT_Int mask = 0, val = 0; in psh_hint_table_activate_mask() local
344 psh_dimension_quantize_len( PSH_Dimension dim, in psh_dimension_quantize_len()
442 PSH_Dimension dim = &globals->dimension[dimension]; in psh_hint_align() local
856 PSH_Dimension dim = &globals->dimension[dimension]; in psh_hint_table_align_hints() local
1570 PSH_Dimension dim = &glyph->globals->dimension[dimension]; in psh_glyph_find_strong_points() local
1715 PSH_Dimension dim = &glyph->globals->dimension[dimension]; in psh_glyph_interpolate_strong_points() local
1770 PSH_Dimension dim = &glyph->globals->dimension[dimension]; in psh_glyph_interpolate_normal_points() local
1920 PSH_Dimension dim = &glyph->globals->dimension[dimension]; in psh_glyph_interpolate_other_points() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | convert_tensor.cc | 109 const auto& val = arr(i); in ConvertStringTensor() local
224 for (auto dim : input_tensor_shape) original_dimensions.push_back(dim.size); in ConvertTensorProto() local
282 for (const auto& dim : shape.dim()) { in ConvertTensorShapeProto() local
293 for (const auto& val : attr.getRawStringData()) in ConvertStringElementsAttr() local
300 for (const auto& val : attr.getValues<std::complex<T>>()) { in ConvertComplexElementsAttr() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | copy_ops_util.cpp | 49 for (const auto& val : stride) { in check_as_strided_copy_args() local
76 int64_t dim, in check_cat_args()
127 int64_t dim, in get_cat_out_target_size()
238 size_t dim = dims[i] >= 0 ? dims[i] : in.dim() + dims[i]; in check_permute_copy_args() local
253 bool check_unbind_copy_args(const Tensor& in, int64_t dim, TensorList out) { in check_unbind_copy_args()
390 int64_t dim, in check_select_copy_out_args()
402 int64_t dim, in get_select_copy_out_target_size()
419 int64_t dim, in check_split_with_sizes_copy_args()
446 int64_t dim, in get_split_with_sizes_copy_out_target_size()
459 int64_t dim, in check_squeeze_copy_dim_args()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_normalization.cc | 99 [&](int dim) { in HandleConcatenate()
115 [&](int dim) { in HandleConcatenate()
129 [&](int dim) { return dim == 1; }); in HandleConcatenate()
392 [&](int dim) { in HandlePad()
412 for (int dim = 0; dim < s_normalized.dimensions_size(); dim++) { in HandlePad() local
416 for (int dim = 0; dim < s.dimensions_size(); dim++) { in HandlePad() local
448 for (int64_t dim : hlo->dimensions()) { in TransformDimensionsForLayoutPreservingHlo() local
464 int DegenDimsUpTo(const Shape& shape, int dim) { in DegenDimsUpTo()
477 int64_t val = dimensions[i]; in NoDegenerateDims() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ReduceOpsKernel.cpp | 33 int64_t dim, in cpu_cum_base_kernel()
79 static void cumsum_cpu_kernel(const Tensor& result, const Tensor& self, int64_t dim) { in cumsum_cpu_kernel()
98 static void cumprod_cpu_kernel(const Tensor& result, const Tensor& self, int64_t dim) { in cumprod_cpu_kernel()
117 static void logcumsumexp_cpu_kernel(Tensor& result, const Tensor& self, int64_t dim) { in logcumsumexp_cpu_kernel()
194 void norm_kernel_cpu_impl(TensorIterator& iter, const double& val) { in norm_kernel_cpu_impl()
213 double val = 0; in norm_kernel_tensor_iterator_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | lower_general_dot.cc | 49 for (auto dim : leftDims) { in transposeReshape() local
56 for (auto dim : rightDims) { in transposeReshape() local
78 for (auto val : transposePermutation) { in transposeReshape() local
115 Value dim = rewriter.create<GetDimensionSizeOp>( in transposeReshape() local
154 for (auto dim : contractDimsAttr) { in processDotArg() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | convert_tensor.cc | 106 const auto& val = arr(i); in ConvertStringTensor() local
224 for (auto dim : input_tensor_shape) original_dimensions.push_back(dim.size); in ConvertTensorProto() local
283 for (const auto& dim : shape.dim()) { in ConvertTensorShapeProto() local
293 for (const auto& val : attr.getRawStringData()) in ConvertStringElementsAttr() local
300 for (const auto& val : attr.getValues<std::complex<T>>()) { in ConvertComplexElementsAttr() local
|
| /aosp_15_r20/external/mesa3d/src/nouveau/codegen/ |
| H A D | nv50_ir_lowering_nv50.cpp | 783 Value *src[3], *val; in handleTEX() local
867 ImmediateValue val; in handleTEX() local
1003 const int dim = i->tex.target.getDim() + i->tex.target.isCube(); in handleTXD() local
1014 Value *src[3], *val; in handleTXD() local
1116 const int dim = suq->tex.target.getDim(); in handleSUQ() local
1667 const int dim = su->tex.target.getDim(); in processSurfaceCoords() local
1923 const int dim = su->tex.target.getDim(); in handleSUREDP() local
1949 const int dim = su->tex.target.getDim(); in handleSUSTP() local
2112 LValue *val = bld.getScratch(); in handlePFETCH() local
|
| H A D | nv50_ir_inlines.h | 170 const ValueRef *ValueRef::getIndirect(int dim) const in getIndirect()
254 Value *Instruction::getIndirect(int s, int dim) const in getIndirect()
264 void Instruction::setFlagsDef(int d, Value *val) in setFlagsDef()
278 void Instruction::setFlagsSrc(int s, Value *val) in setFlagsSrc()
|
| H A D | nv50_ir_lowering_nvc0.cpp | 933 const int dim = i->tex.target.getDim() + i->tex.target.isCube(); in handleTEX() local
942 Value *src[3], *val; in handleTEX() local
1170 ImmediateValue val; in handleTEX() local
1221 const int dim = i->tex.target.getDim() + i->tex.target.isCube(); in handleManualTXD() local
1243 Value *src[3], *val; in handleManualTXD() local
1320 int dim = txd->tex.target.getDim() + txd->tex.target.isCube(); in handleTXD() local
1938 int dim = suq->tex.target.getDim(); in handleSUQ() local
2027 const int dim = su->tex.target.getDim(); in processSurfaceCoordsNVE4() local
2424 const int dim = su->tex.target.getDim(); in processSurfaceCoordsNVC0() local
2587 const int dim = su->tex.target.getDim(); in handleSurfaceOpNVC0() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | shape_inference.h | 136 DimensionHandle(const Dimension* dim) { ptr_ = dim; } in DimensionHandle()
201 DimensionHandle dim; member
202 int64_t val; member
882 inline DimensionOrConstant::DimensionOrConstant(DimensionHandle dim) in DimensionOrConstant()
887 inline DimensionOrConstant::DimensionOrConstant(int64_t val) : val(val) { in DimensionOrConstant()
|
| /aosp_15_r20/prebuilts/devtools/tools/lib/ |
| HD | traceview.jar | ... {
final synthetic com.android.traceview.TimeBase val$timeBase
final synthetic com.android.traceview. ... |
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_special_ops.cpp | 78 void checkSequenceSize(int64_t n, int64_t dim, int64_t seq_size) { in checkSequenceSize()
96 int64_t dim, in storeLastDimension()
112 int64_t dim, in storeLastDimensionFloat()
128 int64_t dim, in storeLastDimensionHalf()
145 int64_t dim, in recursiveStore()
435 double val = 0; in __anonedd36e380c02() local
|