Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dtriangular_solve_thunk.cc41 int64_t a_batch_stride, int64_t b_batch_stride) in TriangularSolveThunk()
92 int64_t a_batch_stride, int64_t b_batch_stride, in RunTriangulatSolve()
H A Djitrt_custom_calls.cc1445 int64_t a_batch_stride = left_side ? m * m * elem_size : n * n * elem_size; in operator ()() local
H A Dir_emitter_unnested.cc1467 int64_t a_batch_stride = in EmitTriangularSolveCustomCall() local