Searched refs:ConvertInfeedtLayout (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 1890 LogicalResult ConvertInfeedtLayout(mlir::Operation* op, in ConvertInfeedtLayout() function
1940 ConvertInfeedtLayout(op, layout, data_subshapes->Mutable(i), i))) in ConvertInfeedtLayout()
1954 if (failed(ConvertInfeedtLayout(op, layout, subshapes->Mutable(i), i))) in ConvertInfeedtLayout()
2084 if (failed(ConvertInfeedtLayout(inst, layout, shape))) in Lower()
2090 if (failed(ConvertInfeedtLayout(inst, layout, shape, i))) in Lower()
2105 if (failed(ConvertInfeedtLayout(inst, layout, data_tuple_shape))) in Lower()
|