Searched refs:set2_st (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | set_kernels.cc | 484 sparse::SparseTensor set2_st; in ComputeDenseToSparse() local
486 SparseTensorFromContext(ctx, 1, validate_indices_, &set2_st)); in ComputeDenseToSparse()
487 OP_REQUIRES_OK(ctx, set2_st.IndicesValid()); in ComputeDenseToSparse()
493 set2_st.shape(), &group_shape)); in ComputeDenseToSparse()
504 set2_st.group(set2_st.order().subspan(0, set2_st.order().size() - 1)); in ComputeDenseToSparse()
535 PopulateFromSparseGroup<T>(ctx, group, set2_st.shape(), in ComputeDenseToSparse()
604 sparse::SparseTensor set2_st; in ComputeSparseToSparse() local
606 SparseTensorFromContext(ctx, 3, validate_indices_, &set2_st)); in ComputeSparseToSparse()
612 OP_REQUIRES_OK(ctx, GroupShapeFromInputs(set1_st.shape(), set2_st.shape(), in ComputeSparseToSparse()
616 const ShapeArray set2_strides = Strides(set2_st.shape()); in ComputeSparseToSparse()
[all …]
|