Searched refs:transformed_filter (Results 1 – 5 of 5) sorted by relevance
281 Tensor transformed_filter; in operator ()() local294 &transformed_filter)); in operator ()()298 To32Bit(transformed_filter.tensor<T, 4>())); in operator ()()350 AsDeviceMemory(transformed_filter.template flat<T>().data(), in operator ()()351 transformed_filter.template flat<T>().size()); in operator ()()
430 Tensor transformed_filter; in launch() local444 &transformed_filter)); in launch()451 To32Bit(transformed_filter.tensor<T, 5>())); in launch()470 AsDeviceMemory(transformed_filter.template flat<T>().data(), in launch()471 transformed_filter.template flat<T>().size()); in launch()
604 Tensor transformed_filter;610 &transformed_filter));614 To32Bit(transformed_filter.tensor<T, 4>()));635 se::DeviceMemory<T> filter_ptr = tensor_on_device(transformed_filter);
906 Tensor transformed_filter; in operator ()() local920 &transformed_filter)); in operator ()()924 To32Bit(transformed_filter.tensor<T, 4>())); in operator ()()954 AsDeviceMemory(transformed_filter.template flat<T>().data(), in operator ()()955 transformed_filter.template flat<T>().size()); in operator ()()
1332 Tensor transformed_filter; in Compute() local1345 &transformed_filter)); in Compute()1350 To32Bit(transformed_filter.tensor<T, 5>())); in Compute()1388 AsDeviceMemory(transformed_filter.template flat<T>().data(), in Compute()1389 transformed_filter.template flat<T>().size()); in Compute()