Searched defs:h_in (Results 1 – 14 of 14) sorted by relevance
24 const std::vector<float> &h_in) in test_function()
26 const std::vector<T> &h_in, in test_selection_merge()
20 const std::vector<Tv> &h_in, const int n) in test_extract()
20 const std::vector<Ts> &h_in, const int n) in test_insert()
26 const std::vector<Tv> &h_in, in test_negation()
110 std::vector<cl_float> h_in(num); in TEST_SPIRV_FUNC() local
56 const int h_in = h_beg + h * rate_rows; in DilationKernel() local101 const int h_in = h_beg + h * rate_rows; in DilationBackpropInputKernel() local151 const int h_in = h_beg + h * rate_rows; in DilationBackpropFilterKernel() local
191 const int h_in = h_beg + h * rate_rows; in operator ()() local314 const int h_in = h_beg + h * rate_rows; in operator ()() local442 const int h_in = h_beg + h * rate_rows; in operator ()() local
60 const T* h_in = labels_indices->flat<T>().data(); in DoHistogram() local
128 const int h_in = -padHeight + h * strideHeight + kH * dilationHeight; in C10_LAUNCH_BOUNDS_1() local200 const int h_in = -padHeight + h * strideHeight + kH * dilationHeight; in C10_LAUNCH_BOUNDS_1() local
49 auto h_in = h_out * stride_h - pad_h; in vol2col_kernel() local
45 int64_t h_in = h_out * stride_height - pad_height; in im2col_kernel() local
496 const int h_in = in.shape()[1]; in winograd_output_transform() local
4107 def compute_indices_adaptive_pooling(start_index, end_index, h_in, w_in, h_out, w_out): argument