Searched refs:num_segments_t (Results 1 – 1 of 1) sorted by relevance
367 Tensor num_segments_t = test::AsScalar(100); in TEST() local368 op.input_tensors[2] = &num_segments_t; in TEST()371 num_segments_t = test::AsScalar(-1); in TEST()396 Tensor num_segments_t = test::AsScalar(100); in TEST() local397 op.input_tensors[3] = &num_segments_t; in TEST()404 num_segments_t = test::AsScalar(-100); in TEST()405 op.input_tensors[3] = &num_segments_t; in TEST()