MCPcopy Create free account
hub / github.com/MegEngine/MegCC / DecodeTernaryBcastType

Function DecodeTernaryBcastType

compiler/lib/KernelGen/Common/ElemwiseCommon.cpp:228–235  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

226}
227
228std::vector<TensorType> DecodeTernaryBcastType(const BcastType bct_type) {
229 std::vector<TensorType> input_type;
230 input_type.push_back(static_cast<TensorType>((bct_type - VEC_VEC_VEC) / 100));
231 input_type.push_back(
232 static_cast<TensorType>(((bct_type - VEC_VEC_VEC) % 100) / 10));
233 input_type.push_back(static_cast<TensorType>((bct_type - VEC_VEC_VEC) % 10));
234 return input_type;
235}
236
237std::vector<size_t> CalBroadcastElemwise(
238 const std::vector<size_t>& src, const std::vector<size_t>& dst) {

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected