Commitf095c05

committed

Supporting scalar tensor broadcasting

1 parentb5c1702 commitf095c05Copy full SHA for f095c05

File tree

-1

lines changed

-1

lines changed

Lines changed: 4 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -278,8 +278,11 @@ void Add(S_TENSOR input, S_TENSOR input2, S_TENSOR out) {`
`278`	`278`	`TOut* p_out = out->write<TOut>(0,0);`
`279`	`279`
`280`	`280`	`constuint32_t size = out->getSize();`
	`281`	`+// [FIXME] hacking for broadcasting scalar tensor`
	`282`	`+constuint32_t size1 = input->getSize();`
	`283`	`+constuint32_t size2 = input2->getSize();`
`281`	`284`	`for (uint32_t i =0; i < size; i++) {`
`282`		`- p_out[i] = p_in[i] + p_in2[i];`
	`285`	`+ p_out[i] = p_in[size1 >1 ? i :0] + p_in2[size2 >1 ? i :0];`
`283`	`286`	`}`
`284`	`287`	`}`
`285`	`288`

Comments

(0)