Searched refs:CreateDTensorLayoutOp (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | propagate_default_layout.cc | 45 void CreateDTensorLayoutOp(const Layout& layout, mlir::Value input, in CreateDTensorLayoutOp() function
82 CreateDTensorLayoutOp(layout, relayout.output(), type, relayout.getLoc(), in PropagateDTensorLayoutForRelayout()
107 CreateDTensorLayoutOp(layout_or_status.ValueOrDie(), arg, type, in PropagateFunctionArgAttrToLayoutOp()
144 CreateDTensorLayoutOp(result_layout_or_status.ValueOrDie(), return_value, in PropagateFunctionDefaultLayoutAttrToLayoutOp()
|