MCPcopy Create free account
hub / github.com/evilsocket/cake / add3

Method add3

cake-core/src/backends/vulkan/mod.rs:1312–1321  ·  view source on GitHub ↗
(&self, a: &Tensor, b: &Tensor, c: &Tensor)

Source from the content-addressed store, hash-verified

1310 }
1311
1312 fn add3(&self, a: &Tensor, b: &Tensor, c: &Tensor) -> Result<Tensor> {
1313 let n = a.elem_count();
1314 if n > 32768 {
1315 log::debug!("add3 GPU: {n} elements");
1316 self.dispatch_ternary_vec4(a, b, c, "add3")
1317 } else {
1318 log::debug!("add3 CPU: {n} elements (<=32768)");
1319 ((a + b)? + c)?.contiguous()
1320 }
1321 }
1322
1323 fn exp_mul(&self, x: &Tensor, y: &Tensor) -> Result<Tensor> {
1324 let n = x.elem_count();

Callers

nothing calls this directly

Calls 1

dispatch_ternary_vec4Method · 0.80

Tested by

no test coverage detected