Searched defs:output_columns (Results 1 – 6 of 6) sorted by relevance
53 auto output_columns = output_shape[1]; in unpack() local127 const int32_t output_columns = input_columns - 2 * sizeof(float); in qembeddingbag_byte_unpack_out() local176 const auto output_columns = input_columns - 2 * sizeof(float); in qembeddingbag_byte_unpack_meta() local215 auto output_columns = output_dimensions[1]; in _qembeddingbag_nbit_unpack_helper() local
90 size_t output_columns = output_shape[1]; in prepack() local246 const int32_t output_columns = embedding_cols + 2 * sizeof(float); in qembeddingbag_byte_prepack_out() local338 const int32_t output_columns = embedding_cols + 2 * sizeof(float); in qembeddingbag_byte_prepack_meta() local430 const auto output_columns = output.size(output.dim() - 1); in _qembeddingbag_nbit_prepack_helper() local
65 const size_t output_columns = ncols; in _float_to_bfloat16_cuda() local116 const size_t output_columns = ncols; in _bfloat16_to_float_cuda() local
933 inline const ::google::internal::federated::plan::ColumnSchema& SqlQuery::output_columns(int index)… in output_columns() function946 SqlQuery::output_columns() const { in output_columns() function
39 repeated ColumnSchema output_columns = 3; field
177 const int32_t output_columns = input_columns - 2 * sizeof(float); in qembeddingbag_byte_unpack() local