Searched refs:entry_weights (Results 1 – 1 of 1) sorted by relevance
79 const Tensor& entry_weights = context->input(6); in Compute() local99 OP_REQUIRES(context, TensorShapeUtils::IsVector(entry_weights.shape()), in Compute()113 entry_weights.dim_size(0) == 0) || in Compute()115 entry_weights.dim_size(0) == input_indices.dim_size(0))), in Compute()130 const auto& entry_weights_vec = entry_weights.vec<float>(); in Compute()