Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dtridiagonal_test.cc90 auto b_i = Coefficient(main_diagonal_xla, i); in XLA_TEST_P() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc574 const T* b_i = b_base_ptr + i * b_slice_size; in Compute() local
662 typename TTypes<T>::UnalignedConstMatrix b_i(b.data() + i * b_slice_size, in Compute() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc936 for (int b_i = 0, in CountBatchMatMulOperations() local
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1145 auto b_i = EmitExtractImag(rhs_value); in EmitComplexBinaryOp() local