Searched refs:pre_transformed_in_backprop (Results 1 – 2 of 2) sorted by relevance
334 Tensor pre_transformed_in_backprop; in operator ()() local344 &pre_transformed_in_backprop)); in operator ()()353 AsDeviceMemory(pre_transformed_in_backprop.template flat<T>().data(), in operator ()()354 pre_transformed_in_backprop.template flat<T>().size()); in operator ()()529 To32Bit(const_cast<const Tensor&>(pre_transformed_in_backprop) in operator ()()537 pre_transformed_in_backprop = in_backprop_remove_padding; in operator ()()545 toConstTensor(pre_transformed_in_backprop).template tensor<T, 4>(), in operator ()()548 *in_backprop = pre_transformed_in_backprop; in operator ()()
1372 Tensor pre_transformed_in_backprop; in Compute() local1382 &pre_transformed_in_backprop)); in Compute()1391 AsDeviceMemory(pre_transformed_in_backprop.template flat<T>().data(), in Compute()1392 pre_transformed_in_backprop.template flat<T>().size()); in Compute()1546 To32Bit(const_cast<const Tensor&>(pre_transformed_in_backprop) in Compute()1552 pre_transformed_in_backprop = in_backprop_remove_padding; in Compute()1559 toConstTensor(pre_transformed_in_backprop).template tensor<T, 5>(), in Compute()1562 *in_backprop = pre_transformed_in_backprop; in Compute()