Searched defs:ComputeTensorSlice (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ | ||
| H A D | lu_op.cc | 128 void ComputeTensorSlice(OpKernelContext* context, int64_t matrix_index, in ComputeTensorSlice() function in tensorflow::LuOp |
| H A D | linalg_ops_common.cc | 232 void LinearAlgebraOp<InputScalar, OutputScalar>::ComputeTensorSlice( in ComputeTensorSlice() function in tensorflow::LinearAlgebraOp |