Searched refs:transformed_filter (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 382 Tensor transformed_filter; in launch() local
387 &transformed_filter)); in launch()
393 To32Bit(transformed_filter.tensor<T, 5>())); in launch()
406 AsDeviceMemory(transformed_filter.template flat<T>().data(), in launch()
407 transformed_filter.template flat<T>().size()); in launch()
|
| D | conv_ops.cc | 789 Tensor transformed_filter; in operator ()() local
794 &transformed_filter)); in operator ()()
798 To32Bit(transformed_filter.tensor<T, 4>())); in operator ()()
815 AsDeviceMemory(transformed_filter.template flat<T>().data(), in operator ()()
816 transformed_filter.template flat<T>().size()); in operator ()()
888 LogConvAutotuneResults(ctx->op_kernel().def(), input, transformed_filter, in operator ()()
|
| D | conv_ops_fused_impl.h | 710 Tensor transformed_filter;
716 &transformed_filter));
720 To32Bit(transformed_filter.tensor<T, 4>()));
741 se::DeviceMemory<T> filter_ptr = tensor_on_device(transformed_filter);
799 context->op_kernel().def(), input, transformed_filter,
|
| D | conv_grad_input_ops.cc | 871 Tensor transformed_filter; in operator ()() local
877 &transformed_filter)); in operator ()()
882 To32Bit(transformed_filter.tensor<T, 4>())); in operator ()()
920 AsDeviceMemory(transformed_filter.template flat<T>().data(), in operator ()()
921 transformed_filter.template flat<T>().size()); in operator ()()
988 transformed_filter, transformed_out_backprop, in operator ()()
|
| D | conv_grad_ops_3d.cc | 1286 Tensor transformed_filter; in Compute() local
1293 &transformed_filter)); in Compute()
1297 To32Bit(transformed_filter.tensor<T, 5>())); in Compute()
1329 AsDeviceMemory(transformed_filter.template flat<T>().data(), in Compute()
1330 transformed_filter.template flat<T>().size()); in Compute()
|