Searched refs:casted_out_backprop (Results 1 – 1 of 1) sorted by relevance
1186 Tensor casted_out_backprop = out_backprop; in Compute() local1194 &casted_out_backprop)); in Compute()1195 cast(device, casted_out_backprop.template flat<float>(), in Compute()1206 auto out_backprop_ptr = casted_out_backprop.template flat<U>().data(); in Compute()