Searched refs:set1_st (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | set_kernels.cc | 599 sparse::SparseTensor set1_st; in ComputeSparseToSparse() local
601 SparseTensorFromContext(ctx, 0, validate_indices_, &set1_st)); in ComputeSparseToSparse()
602 OP_REQUIRES_OK(ctx, set1_st.IndicesValid()); in ComputeSparseToSparse()
612 OP_REQUIRES_OK(ctx, GroupShapeFromInputs(set1_st.shape(), set2_st.shape(), in ComputeSparseToSparse()
615 const ShapeArray set1_strides = Strides(set1_st.shape()); in ComputeSparseToSparse()
625 set1_st.group(set1_st.order().subspan(0, set1_st.order().size() - 1)); in ComputeSparseToSparse()
649 PopulateFromSparseGroup<T>(ctx, *set1_group_it, set1_st.shape(), in ComputeSparseToSparse()
|