Lines Matching defs:partitions
43 const Tensor** partitions,
46 OP_REQUIRES_OK(c, c->input("partitions", partitions));
49 TensorShapeUtils::StartsWith((*data)->shape(), (*partitions)->shape()),
51 "data.shape must start with partitions.shape, ",
53 ", partitions.shape = ", (*partitions)->shape().DebugString()));
55 // Count how many occurrences of each partition id we have in partitions
57 auto e_partitions = (*partitions)->flat<int32>();
63 "partitions", SliceDebugString((*partitions)->shape(), i),
73 for (int i = (*partitions)->dims(); i < (*data)->dims(); i++) {
92 const Tensor* partitions;
94 ValidateAndAllocateOutputs(c, &data, &partitions, &outputs);
98 auto e_partitions = partitions->flat<int32>();
102 if (partitions->dims() == data->dims()) {
