Searched refs:matrix_index (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | 219 OpKernelContext* context, int64 matrix_index, const TensorInputs& inputs, in ComputeTensorSlice() argument
228 matrix_index * input_matrix_shapes[i].num_elements(), in ComputeTensorSlice()
243 matrix_index * output_matrix_shapes[i].num_elements(), in ComputeTensorSlice()
|
| D | linalg_ops_common.h | 150 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()
|