Searched refs:input_st (Results 1 – 2 of 2) sorted by relevance
193 SparseTensor input_st; in Compute() local196 &input_st)); in Compute()204 OP_REQUIRES_OK(context, input_st.IndicesValid()); in Compute()227 sparse::GroupIterable minibatch = input_st.group({0}); in Compute()
259 SparseTensor input_st; in Compute() local262 &input_st)); in Compute()270 OP_REQUIRES_OK(context, input_st.IndicesValid()); in Compute()281 sparse::GroupIterable minibatch = input_st.group({0}); in Compute()