Searched refs:ComputeTensorSlice (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | lu_op.cc | 118 ComputeTensorSlice(context, i, input, num_rows, num_cols, outputs, in Compute()
128 void ComputeTensorSlice(OpKernelContext* context, int64_t matrix_index, in ComputeTensorSlice() function in tensorflow::LuOp
|
| H A D | linalg_ops_common.cc | 110 ComputeTensorSlice(context, i, inputs, input_matrix_shapes, outputs, in Compute()
232 void LinearAlgebraOp<InputScalar, OutputScalar>::ComputeTensorSlice( in ComputeTensorSlice() function in tensorflow::LinearAlgebraOp
|
| H A D | linalg_ops_common.h | 174 void ComputeTensorSlice(OpKernelContext* context, int64_t matrix_index,
|