Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_band_part_op.cc74 const int64_t num_lower = as_int64_scalar(num_lower_in); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc191 auto num_lower = op.input(1); in MatrixBandPartGrad() local
H A Darray_grad_test.cc154 const int64_t num_lower = 1; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc663 static RankedTensorType Get2DTensorType(Value input, Value num_lower) { in Get2DTensorType()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc3266 auto num_lower = in TEST_F() local