Searched refs:kernel_multi_index (Results 1 – 1 of 1) sorted by relevance
1065 std::vector<llvm::Value*> kernel_multi_index(num_dims); in EmitElementalConvolution() local1067 kernel_multi_index[dnums.kernel_spatial_dimensions(i)] = in EmitElementalConvolution()1074 kernel_multi_index[dnums.kernel_input_feature_dimension()] = input_feature; in EmitElementalConvolution()1075 kernel_multi_index[dnums.kernel_output_feature_dimension()] = output_feature; in EmitElementalConvolution()1081 llvm_ir::IrArray::Index kernel_index(kernel_multi_index, rhs->shape(), in EmitElementalConvolution()