Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_band_part_op.cc85 const int64_t num_upper = as_int64_scalar(num_upper_in); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc192 auto num_upper = op.input(2); in MatrixBandPartGrad() local
H A Darray_grad_test.cc155 const int64_t num_upper = 2; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc3269 auto num_upper = in TEST_F() local