Searched refs:kernel_multi_index (Results 1 – 1 of 1) sorted by relevance
3148 std::vector<llvm::Value*> kernel_multi_index(num_dims); in EmitConvolution() local3150 kernel_multi_index[dnums.kernel_spatial_dimensions(i)] = in EmitConvolution()3157 kernel_multi_index[dnums.kernel_input_feature_dimension()] = input_feature; in EmitConvolution()3158 kernel_multi_index[dnums.kernel_output_feature_dimension()] = output_feature; in EmitConvolution()3164 llvm_ir::IrArray::Index kernel_index(kernel_multi_index, rhs->shape(), in EmitConvolution()