Searched refs:pre_transformed_in_backprop (Results 1 – 2 of 2) sorted by relevance
904 Tensor pre_transformed_in_backprop; in operator ()() local914 &pre_transformed_in_backprop)); in operator ()()923 AsDeviceMemory(pre_transformed_in_backprop.template flat<T>().data(), in operator ()()924 pre_transformed_in_backprop.template flat<T>().size()); in operator ()()987 LogConvAutotuneResults(ctx->op_kernel().def(), pre_transformed_in_backprop, in operator ()()1029 To32Bit(const_cast<const Tensor&>(pre_transformed_in_backprop) in operator ()()1036 pre_transformed_in_backprop = in_backprop_remove_padding; in operator ()()1043 toConstTensor(pre_transformed_in_backprop).template tensor<T, 4>(), in operator ()()1046 *in_backprop = pre_transformed_in_backprop; in operator ()()
1319 Tensor pre_transformed_in_backprop; in Compute() local1323 &pre_transformed_in_backprop)); in Compute()1332 AsDeviceMemory(pre_transformed_in_backprop.template flat<T>().data(), in Compute()1333 pre_transformed_in_backprop.template flat<T>().size()); in Compute()1436 To32Bit(const_cast<const Tensor&>(pre_transformed_in_backprop) in Compute()1441 pre_transformed_in_backprop = in_backprop_remove_padding; in Compute()1448 toConstTensor(pre_transformed_in_backprop).template tensor<T, 5>(), in Compute()1451 *in_backprop = pre_transformed_in_backprop; in Compute()