Searched refs:transformed_output (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 60 Tensor transformed_output; in Compute() local
66 &transformed_output)); in Compute()
68 transformed_output = *output; in Compute()
95 AsDeviceMemory(transformed_output.template flat<T>().data(), in Compute()
96 transformed_output.template flat<T>().size()); in Compute()
112 toConstTensor(transformed_output).template tensor<T, 5>(), in Compute()
144 Tensor transformed_output; in Compute() local
151 &transformed_output)); in Compute()
153 transformed_output = *tensor_out; in Compute()
182 transformed_output.tensor<T, 5>()); in Compute()
[all …]
|
| D | pooling_ops_common.cc | 192 Tensor transformed_output; in Compute() local
198 &transformed_output)); in Compute()
200 transformed_output = *tensor_out; in Compute()
205 auto& transformed_output = *tensor_out; in Compute() local
259 transformed_output.template flat<T>().data()), in Compute()
260 transformed_output.template flat<T>().size()); in Compute()
278 toConstTensor(transformed_output).template tensor<RT, 4>(), in Compute()
322 Tensor transformed_output; in Compute() local
329 &transformed_output)); in Compute()
331 transformed_output = *tensor_out; in Compute()
[all …]
|
| D | conv_ops_3d.cc | 395 Tensor transformed_output; in launch() local
401 &transformed_output)); in launch()
409 AsDeviceMemory(transformed_output.template flat<T>().data(), in launch()
410 transformed_output.template flat<T>().size()); in launch()
504 const_cast<const Tensor&>(transformed_output).tensor<T, 5>(), in launch()
507 *output = transformed_output; in launch()
|
| D | conv_ops.cc | 800 Tensor transformed_output; in operator ()() local
807 &transformed_output)); in operator ()()
809 transformed_output = *output; in operator ()()
818 AsDeviceMemory(transformed_output.template flat<T>().data(), in operator ()()
819 transformed_output.template flat<T>().size()); in operator ()()
889 transformed_output, stream->parent(), results); in operator ()()
913 const_cast<const Tensor&>(transformed_output).tensor<T, 4>(), in operator ()()
|
| D | conv_ops_fused_impl.h | 722 Tensor transformed_output;
730 &transformed_output));
732 transformed_output = *output;
743 se::DeviceMemory<T> output_ptr = tensor_on_device(transformed_output);
800 transformed_output, bias, nullptr, stream->parent(), results);
819 const_cast<const Tensor&>(transformed_output).tensor<T, 4>(),
|