Searched defs:num_upper (Results 1 – 4 of 4) sorted by relevance
52 xla::XlaOp num_upper = context->Input(2); in Compile() local
84 const int64 num_upper = as_int64_scalar(num_upper_in); in Compute() local
169 auto num_upper = op.input(2); in MatrixBandPartGrad() local
147 const int64 num_upper = 2; in TEST_F() local