Searched refs:input_transposed (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | qr_op_impl.h | 177 Tensor input_transposed; in ComputeAsync() local
185 transposed_shape, &input_transposed), in ComputeAsync()
199 context, DoMatrixTranspose(device, input, &input_transposed), done); in ComputeAsync()
205 input_transposed.flat_inner_dims<Scalar, 3>(); in ComputeAsync()
222 context, DoMatrixTranspose(device, input_transposed, r), done); in ComputeAsync()
284 context, DoMatrixTranspose(device, input_transposed, q), done); in ComputeAsync()
|
| D | einsum_op_impl.h | 464 Tensor input_transposed; in ReduceOperand() local
480 &input_transposed)); in ReduceOperand()
487 StrideOrInflate<Device, T>(ctx, input_transposed, *labels, label_counts, in ReduceOperand()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_test.cc | 62 std::vector<integer_type>* input_transposed) { in ComputeExpectedTransposeResult() argument
66 input_transposed->resize(count); in ComputeExpectedTransposeResult()
89 input_transposed->data()); in ComputeExpectedTransposeResult()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_test.cc | 38 std::vector<T>* input_transposed) { in RunTestPermutation() argument
42 input_transposed->resize(count); in RunTestPermutation()
64 input_transposed->data()); in RunTestPermutation()
|