Searched refs:input_st (Results 1 – 2 of 2) sorted by relevance
370 SparseTensor input_st; in Compute() local373 &input_st)); in Compute()382 OP_REQUIRES_OK(context, input_st.IndicesValid()); in Compute()389 sparse::GroupIterable minibatch = input_st.group({0}); in Compute()
260 SparseTensor input_st; in Compute() local263 &input_st)); in Compute()271 OP_REQUIRES_OK(context, input_st.IndicesValid()); in Compute()282 sparse::GroupIterable minibatch = input_st.group({0}); in Compute()