Home
last modified time | relevance | path

Searched defs:diagonal_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_diag_op.cc223 const TensorShape& diagonal_shape = diagonal.shape(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dtridiagonal.cc122 Status ValidateTridiagonalMatMulDiagonal(const Shape& diagonal_shape, in ValidateTridiagonalMatMulDiagonal()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc3327 std::vector<int64_t> diagonal_shape(shape); in TEST_F() local
3352 std::vector<int64_t> diagonal_shape(shape); in TEST_F() local