Searched refs:matrix_index (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/linalg/ |
D | lu_op.cc | 128 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()
|
D | linalg_ops_common.cc | 221 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()
|
D | linalg_ops_common.h | 173 void ComputeTensorSlice(OpKernelContext* context, int64 matrix_index,
|
/external/eigen/bench/ |
D | benchmark-blocking-sizes.cpp | 201 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()
|