Searched refs:dense_shape_ (Results 1 – 2 of 2) sorted by relevance
87 dense_shape_(DT_INT64, {params.dataset->sparse_tensor_.dims() - 1}), in Iterator()90 for (size_t i = 0; i < dense_shape_.NumElements(); ++i) {91 dense_shape_.vec<int64>()(i) =140 out_tensors->push_back(dense_shape_); in GetNextInternal()148 out_tensors->push_back(dense_shape_); in GetNextInternal()200 Tensor dense_shape_; member in tensorflow::data::__anone5092be70111::Dataset::Iterator
41 dense_shape_ = AddInput(TensorType_INT32); in EmbeddingLookupSparseOpModel()58 PopulateTensor(dense_shape_, dense_shape_data); in SetInput()82 int dense_shape_; member in tflite::__anonb811566d0111::EmbeddingLookupSparseOpModel