HomeSort by relevance Sort by last modified time
    Searched refs:b_batch_stride (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
triangular_solve_thunk.h 45 int64 a_batch_stride, int64 b_batch_stride,
triangular_solve_thunk.cc 39 int64 b_batch_stride, const HloInstruction* hlo)
54 b_batch_stride_(b_batch_stride) {
79 << " b_batch_stride=" << b_batch_stride_;
ir_emitter_unnested.cc 1898 int64 b_batch_stride = m * n * elem_size; local
    [all...]

Completed in 190 milliseconds