Searched refs:Transpose2D (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 109 Value Transpose2D(OpBuilder* builder, Value value_to_transpose, in Transpose2D() function
342 Transpose2D(&builder_, weight_, weight_type_, fused_func_op_.getLoc()); in GenerateFusedOpOperands()
343 projection_transposed_ = Transpose2D(&builder_, projection_, projection_type_, in GenerateFusedOpOperands()
678 Transpose2D(builder, weight_kernel, weight_type, func_op.getLoc()); in ConvertKerasLSTMLayer()
684 Value transpose_recurrent_kernel = Transpose2D( in ConvertKerasLSTMLayer()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7597 inline void Transpose2D(const RuntimeShape& input_shape, const T* input_data, in Transpose2D() function
7693 inline void Transpose2D(const RuntimeShape& input_shape, in Transpose2D() function
7832 Transpose2D(RuntimeShape({dim0, dim1}), input_data, in TransposeImpl()
|