Home
last modified time | relevance | path

Searched defs:b_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.cc1446 int64_t b_batch_stride = m * n * elem_size; in operator ()() local
H A Dir_emitter_unnested.cc1469 int64_t b_batch_stride = m * n * elem_size; in EmitTriangularSolveCustomCall() local