Home
last modified time | relevance | path

Searched refs:matrix_index (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op.cc128 void ComputeTensorSlice(OpKernelContext* context, int64 matrix_index, in ComputeTensorSlice() argument
135 input.flat<Scalar>().data() + matrix_index * num_rows * num_cols, in ComputeTensorSlice()
140 outputs[0]->flat<Scalar>().data() + matrix_index * num_rows * num_cols, in ComputeTensorSlice()
145 outputs[1]->flat<Tidx>().data() + matrix_index * num_rows, num_rows, 1); in ComputeTensorSlice()
Dlinalg_ops_common.cc221 OpKernelContext* context, int64 matrix_index, const TensorInputs& inputs, in ComputeTensorSlice() argument
230 matrix_index * input_matrix_shapes[i].num_elements(), in ComputeTensorSlice()
245 matrix_index * output_matrix_shapes[i].num_elements(), in ComputeTensorSlice()
Dlinalg_ops_common.h173 void ComputeTensorSlice(OpKernelContext* context, int64 matrix_index,
/external/eigen/bench/
Dbenchmark-blocking-sizes.cpp201 size_t matrix_index = 0; in run() local
206 dst[matrix_index].noalias() = lhs[matrix_index] * rhs[matrix_index]; in run()
207 matrix_index++; in run()
208 if (matrix_index == matrix_pool_size) { in run()
209 matrix_index = 0; in run()