Searched defs:TensorSetZero (Results 1 – 2 of 2) sorted by relevance
25 struct TensorSetZero { struct26 void operator()(const Device& d, typename TTypes<T>::Flat t) { in operator()
73 Status TensorSetZero(OpKernelContext* ctx, Tensor* value) { in TensorSetZero() function