Searched refs:row_splits_out (Results 1 – 2 of 2) sorted by relevance
358 Tensor* row_splits_out; in Compute() local360 &values_out, &row_splits_out)); in Compute()363 features, num_buckets_, hash_key_, row_splits_out, values_out); in Compute()591 Tensor** row_splits_out) { in BuildOutputTensors() argument594 1, TensorShape({batch_size + 1}), row_splits_out)); in BuildOutputTensors()595 auto flat_row_splits = (*row_splits_out)->flat<SplitsType>(); in BuildOutputTensors()
1352 int64* row_splits_out = &row_splits->flat<int64_t>()(splits_offset); in FastParseExample() local1353 int64_t start = *row_splits_out; in FastParseExample()1355 *++row_splits_out = start + example_end_index; in FastParseExample()1358 int32* row_splits_out = &row_splits->flat<int32>()(splits_offset); in FastParseExample() local1359 int32_t start = *row_splits_out; in FastParseExample()1361 *++row_splits_out = start + example_end_index; in FastParseExample()