Searched refs:SparseTensorSliceDatasetParams (Results 1 – 1 of 1) sorted by relevance
30 class SparseTensorSliceDatasetParams : public DatasetParams { class32 SparseTensorSliceDatasetParams(Tensor indices, Tensor values, in SparseTensorSliceDatasetParams() function in tensorflow::data::__anon53ef85d20111::SparseTensorSliceDatasetParams73 SparseTensorSliceDatasetParams TwoDimsSparseTensorSliceDatasetParams() { in TwoDimsSparseTensorSliceDatasetParams()74 return SparseTensorSliceDatasetParams( in TwoDimsSparseTensorSliceDatasetParams()82 SparseTensorSliceDatasetParams ThreeDimsSparseTensorSliceDatasetParams() { in ThreeDimsSparseTensorSliceDatasetParams()83 return SparseTensorSliceDatasetParams( in ThreeDimsSparseTensorSliceDatasetParams()91 SparseTensorSliceDatasetParams FourDimsSparseTensorSliceDatasetParams() { in FourDimsSparseTensorSliceDatasetParams()92 return SparseTensorSliceDatasetParams( in FourDimsSparseTensorSliceDatasetParams()100 SparseTensorSliceDatasetParams FiveDimsSparseTensorSliceDatasetParams() { in FiveDimsSparseTensorSliceDatasetParams()101 return SparseTensorSliceDatasetParams( in FiveDimsSparseTensorSliceDatasetParams()[all …]