Home
last modified time | relevance | path

Searched refs:b_base_ptr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h331 auto* b_base_ptr = in_y.template flat<Scalar>().data();
337 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));
350 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));
467 auto* b_base_ptr = in_y.template flat<Scalar>().data();
473 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));
486 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));
Dmatrix_triangular_solve_op_impl.h280 auto* b_base_ptr = in_y.template flat<Scalar>().data();
283 b_tmp_ptrs.push_back(b_base_ptr + i * m * n);
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc547 const T* b_base_ptr = b_t.template flat<T>().data(); in Compute() local
562 b_base_ptr = b_conj_t.template flat<T>().data(); in Compute()
574 const T* b_i = b_base_ptr + i * b_slice_size; in Compute()