Searched refs:splits_shape (Results 1 – 3 of 3) sorted by relevance
303 ShapeHandle splits_shape = c->input(1); in __anon4354d5f10802() local304 TF_RETURN_IF_ERROR(c->WithRank(splits_shape, 1, &unused)); in __anon4354d5f10802()308 TF_RETURN_IF_ERROR(c->Subtract(c->Dim(splits_shape, 0), 1, &dims[0])); in __anon4354d5f10802()
1286 splits_shape = array_ops.shape(self.row_splits, out_type=out_type)1289 ragged_dimensions = [splits_shape[0] - 1] + [
2411 TensorShape values_shape, splits_shape; in ParseContextRaggedFeatures() local2417 splits_shape.AddDim(num_examples + 1); in ParseContextRaggedFeatures()2421 Tensor(allocator, splits_dtype, splits_shape); in ParseContextRaggedFeatures()