Home
last modified time | relevance | path

Searched refs:w_index (Results 1 – 8 of 8) sorted by relevance

/external/libgav1/libgav1/src/dsp/
Dweight_mask.cc55 #define INIT_WEIGHT_MASK(width, height, bitdepth, w_index, h_index) \ argument
56 dsp->weight_mask[w_index][h_index][0] = \
58 dsp->weight_mask[w_index][h_index][1] = \
/external/llvm/test/CodeGen/AMDGPU/
Dvector-alloca.ll37 define void @vector_write(i32 addrspace(1)* %out, i32 %w_index, i32 %r_index) {
48 %1 = getelementptr [4 x i32], [4 x i32]* %0, i32 0, i32 %w_index
60 define void @bitcast_gep(i32 addrspace(1)* %out, i32 %w_index, i32 %r_index) {
/external/libgav1/libgav1/src/dsp/x86/
Dweight_mask_sse4.cc433 #define INIT_WEIGHT_MASK_8BPP(width, height, w_index, h_index) \ argument
434 dsp->weight_mask[w_index][h_index][0] = \
436 dsp->weight_mask[w_index][h_index][1] = WeightMask##width##x##height##_SSE4<1>
905 #define INIT_WEIGHT_MASK_10BPP(width, height, w_index, h_index) \ argument
906 dsp->weight_mask[w_index][h_index][0] = \
908 dsp->weight_mask[w_index][h_index][1] = \
/external/libgav1/libgav1/src/dsp/arm/
Dweight_mask_neon.cc420 #define INIT_WEIGHT_MASK_8BPP(width, height, w_index, h_index) \ argument
421 dsp->weight_mask[w_index][h_index][0] = \
423 dsp->weight_mask[w_index][h_index][1] = WeightMask##width##x##height##_NEON<1>
/external/llvm-project/llvm/test/CodeGen/AMDGPU/
Dvector-alloca.ll37 ; OPT: %0 = insertelement <4 x i32> zeroinitializer, i32 1, i32 %w_index
48 define amdgpu_kernel void @vector_write(i32 addrspace(1)* %out, i32 %w_index, i32 %r_index) {
59 %tmp1 = getelementptr [4 x i32], [4 x i32] addrspace(5)* %tmp, i32 0, i32 %w_index
75 define amdgpu_kernel void @bitcast_gep(i32 addrspace(1)* %out, i32 %w_index, i32 %r_index) {
Dvector-alloca-bitcast.ll45 ; OPT: %0 = insertelement <4 x i32> zeroinitializer, i32 1, i32 %w_index
57 define amdgpu_kernel void @vector_write_alloca_bitcast(i32 addrspace(1)* %out, i32 %w_index, i32 %r…
68 %tmp1 = getelementptr [4 x i32], [4 x i32] addrspace(5)* %tmp, i32 0, i32 %w_index
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2258 const int w_index) { in GetSpatialDimsFromOutputSizes() argument
2267 output_sizes_values[output_sizes_length == 4 ? w_index : 1]}; in GetSpatialDimsFromOutputSizes()
2301 int w_index = (data_format == "NHWC") ? 2 : 3; in ConvertConv2DHelper() local
2313 const nvinfer1::DimsHW dilation(tf_dilations[h_index], tf_dilations[w_index]); in ConvertConv2DHelper()
2337 const nvinfer1::DimsHW stride(tf_stride[h_index], tf_stride[w_index]); in ConvertConv2DHelper()
2377 GetSpatialDimsFromOutputSizes(backprop_output_size, h_index, w_index); in ConvertConv2DHelper()
2437 GetSpatialDimsFromOutputSizes(backprop_output_size, h_index, w_index); in ConvertConv2DHelper()
3257 const int w_index = is_ndhwc ? 3 : 4; in ConvertConv3DHelper() local
3272 tf_dilations[d_index], tf_dilations[h_index], tf_dilations[w_index]); in ConvertConv3DHelper()
3295 tf_stride[w_index]); in ConvertConv3DHelper()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc365 int w_index = 2; in GetPoolingOutputSize() local
369 w_index = 3; in GetPoolingOutputSize()
375 int w = input[w_index]; in GetPoolingOutputSize()
378 int sy = strides[w_index]; in GetPoolingOutputSize()
380 int ky = ksize[w_index]; in GetPoolingOutputSize()