Searched defs:weight_shape (Results 1 – 3 of 3) sorted by relevance
46 std::initializer_list<uint32_t> weight_shape) in LSHProjectionOpModel()
57 std::initializer_list<uint32_t> weight_shape) { in EmbeddingLookupOpModel()
65 const Shape& weight_shape, const int32_t* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias()