Searched refs:factor_weights (Results 1 – 1 of 1) sorted by relevance
74 const Tensor& factor_weights = context->input(1); in Compute() local85 OP_REQUIRES(context, TensorShapeUtils::IsVector(factor_weights.shape()), in Compute()112 factor_weights.dim_size(0) == factors.dim_size(0) && in Compute()114 (input_weights.dim_size(0) == 0 && factor_weights.dim_size(0) == 0 && in Compute()126 const auto& factor_weights_vec = factor_weights.vec<float>(); in Compute()