Searched refs:dense_shape_shape (Results 1 – 2 of 2) sorted by relevance
37 std::initializer_list<int> dense_shape_shape, in EmbeddingLookupSparseOpModel() argument48 BuildInterpreter({lookup_shape, indices_shape, dense_shape_shape, in EmbeddingLookupSparseOpModel()
132 dense_shape_shape = dense_shape.get_shape().with_rank(1)138 indices_shape.dims[1].merge_with(dense_shape_shape.dims[0])