MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / PyBfloat16_Multiply

Function PyBfloat16_Multiply

tensorflow/compiler/xla/python/bfloat16.cc:233–239  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

231}
232
233PyObject* PyBfloat16_Multiply(PyObject* a, PyObject* b) {
234 bfloat16 x, y;
235 if (SafeCastToBfloat16(a, &x) && SafeCastToBfloat16(b, &y)) {
236 return PyBfloat16_FromBfloat16(x * y).release();
237 }
238 return PyArray_Type.tp_as_number->nb_multiply(a, b);
239}
240
241PyObject* PyBfloat16_TrueDivide(PyObject* a, PyObject* b) {
242 bfloat16 x, y;

Callers

nothing calls this directly

Calls 3

SafeCastToBfloat16Function · 0.85
PyBfloat16_FromBfloat16Function · 0.70
releaseMethod · 0.45

Tested by

no test coverage detected