Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dmatrix_solve_op_test.py54 def testSolve(self, n, nrhs, batch_dims, rhs_batch_dims, adjoint): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_layout_assignment.cc280 absl::Span<const int64_t> rhs_batch_dims = in AddBackendConstraints() local
H A Dmatmul_utils.cc236 absl::Span<const int64_t> rhs_batch_dims, in For()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_parser.cc2754 optional<std::vector<int64_t>> rhs_batch_dims; in CreateInstruction() local