Searched refs:splits_size (Results 1 – 5 of 5) sorted by relevance
59 int64 splits_size = splits.size(); in BuildEncodeRaggedTensorGraph() local60 AddInputFromArray<SPLIT_TYPE>(TensorShape({splits_size}), splits); in BuildEncodeRaggedTensorGraph()72 int splits_size = ragged_split.size(); in CreateVariantFromRagged() local74 TensorShape({splits_size})); in CreateVariantFromRagged()
53 int64 splits_size = splits.size(); in BuildRaggedTensorToSparseGraph() local54 AddInputFromArray<int64>(TensorShape({splits_size}), splits); in BuildRaggedTensorToSparseGraph()
55 int64 splits_size = splits.size(); in BuildRaggedGatherGraph() local56 AddInputFromArray<int64>(TensorShape({splits_size}), splits); in BuildRaggedGatherGraph()
100 int splits_size = ragged_components.size() + 1; in NestedStackRaggedTensors() local102 Tensor(DataTypeToEnum<SPLIT_TYPE>::value, TensorShape({splits_size}))); in NestedStackRaggedTensors()
65 int splits_size = ragged_split.size(); in CreateVariantFromRagged() local67 TensorShape({splits_size})); in CreateVariantFromRagged()