Searched defs:num_lower (Results 1 – 5 of 5) sorted by relevance
51 xla::XlaOp num_lower = context->Input(1); in Compile() local
74 const int64 num_lower = as_int64_scalar(num_lower_in); in Compute() local
177 auto num_lower = op.input(1); in MatrixBandPartGrad() local
146 const int64 num_lower = 1; in TEST_F() local
790 static RankedTensorType Get2DTensorType(Value input, Value num_lower) { in Get2DTensorType()