Home
last modified time | relevance | path

Searched defs:num_lower (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_band_part_op.cc51 xla::XlaOp num_lower = context->Input(1); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dmatrix_band_part_op.cc73 const int64 num_lower = as_int64_scalar(num_lower_in); in Compute() local
/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc168 auto num_lower = op.input(1); in MatrixBandPartGrad() local
Darray_grad_test.cc146 const int64 num_lower = 1; in TEST_F() local