Searched refs:left_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 3893 int left_dim = 1, right_dim = 1; in convertOneHotOp() local
3901 left_dim *= dim; in convertOneHotOp()
3945 RankedTensorType::get({left_dim, right_dim, K}, in convertOneHotOp()
3948 rewriter.getI64ArrayAttr({left_dim, right_dim, K})); in convertOneHotOp()
3958 RankedTensorType::get({left_dim, K, right_dim}, in convertOneHotOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1224 int32 left_dim = right_dim = 1
1231 left_dim *= dim
1241 %op7_reshape_op6 = tosa.RESHAPE(%op6_scatter_op4_op5_op2) {shape={left_dim, right_dim, K}}
|