Searched defs:kernel_size_array (Results 1 – 8 of 8) sorted by relevance
33 inline int64_t ComputeValid(int64_t in_value, const ArrayValue<int64_t> &kernel_size_array, in ComputeValid()128 const auto &kernel_size_array = kernel_size_array_opt.value(); in InferShape() local165 …ckKernelSizeAndStrides(const PrimitivePtr &primitive, const ArrayValue<int64_t> &kernel_size_array, in CheckKernelSizeAndStrides()216 const auto &kernel_size_array = kernel_size_array_opt.value(); in CheckValidation() local
129 const auto &kernel_size_array = kernel_size_array_opt.value(); in InferShape() local181 const auto &kernel_size_array = kernel_size_array_opt.value(); in CheckValidation() local
155 const auto &kernel_size_array = kernel_size_array_opt.value(); in InferShape() local206 const auto &kernel_size_array = kernel_size_array_opt.value(); in CheckValidation() local
187 auto kernel_size_array = GetArrayValue<int64_t>(input_values[kIndex1]).value(); in InferShape() local365 const auto &kernel_size_array = kernel_size_opt.value(); in CheckValidation() local415 auto kernel_size_array = GetArrayValue<int64_t>(input_values[kIndex2]).value(); in InferShape() local
36 auto kernel_size_array = ConvertValueTupleToVector<int64_t>(kernel_size); in MaxPoolWithMaskAscendCall() local
36 auto kernel_size_array = ConvertValueTupleToVector<int64_t>(kernel_size); in MaxPoolWithIndicesAscendCall() local
36 auto kernel_size_array = ConvertValueTupleToVector<int64_t>(kernel_size); in MaxPoolGradWithMaskAscendCall() local
37 auto kernel_size_array = ConvertValueTupleToVector<int64_t>(kernel_size); in MaxPoolGradWithIndicesAscendCall() local