Searched refs:transformed_output (Results 1 – 6 of 6) 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()
127 toConstTensor(transformed_output).template tensor<T, 5>(), in Compute()
159 Tensor transformed_output; in Compute() local
166 &transformed_output)); in Compute()
168 transformed_output = *tensor_out; in Compute()
197 transformed_output.tensor<T, 5>()); in Compute()
[all …]
|
| D | pooling_ops_common.cc | 256 Tensor transformed_output; in Compute() local
262 &transformed_output)); in Compute()
264 transformed_output = *tensor_out; in Compute()
269 auto& transformed_output = *tensor_out; in Compute() local
383 transformed_output.template flat<T>().data()), in Compute()
384 transformed_output.template flat<T>().size()); in Compute()
416 toConstTensor(transformed_output).template tensor<RT, 4>(), in Compute()
482 Tensor transformed_output; in Compute() local
489 &transformed_output)); in Compute()
491 transformed_output = *tensor_out; in Compute()
[all …]
|
| D | lrn_op.cc | 249 Tensor transformed_output; in launch()
254 &transformed_output)); in launch()
273 AsDeviceMemory(transformed_output.template flat<T>().data(), in launch()
274 transformed_output.template flat<T>().size()); in launch()
291 toConstTensor(transformed_output).template tensor<T, 4>(), in launch()
563 Tensor transformed_output; in launch()
568 &transformed_output)); in launch()
593 AsDeviceMemory(transformed_output.template flat<T>().data(), in launch()
594 transformed_output.template flat<T>().size()); in launch()
621 toConstTensor(transformed_output).template tensor<T, 4>(), in launch()
|
| D | conv_ops_3d.cc | 453 Tensor transformed_output; in launch() local
462 &transformed_output)); in launch()
464 transformed_output = *output; in launch()
473 AsDeviceMemory(transformed_output.template flat<T>().data(), in launch()
474 transformed_output.template flat<T>().size()); in launch()
639 const_cast<const Tensor&>(transformed_output).tensor<T, 5>(), in launch()
|
| D | conv_ops_fused_impl.h | 616 Tensor transformed_output;
624 &transformed_output));
626 transformed_output = *output;
637 se::DeviceMemory<T> output_ptr = tensor_on_device(transformed_output);
709 const_cast<const Tensor&>(transformed_output).tensor<T, 4>(),
|
| D | conv_ops.cc | 939 Tensor transformed_output; in operator ()() local
946 &transformed_output)); in operator ()()
948 transformed_output = *output; in operator ()()
957 AsDeviceMemory(transformed_output.template flat<T>().data(), in operator ()()
958 transformed_output.template flat<T>().size()); in operator ()()
1128 const_cast<const Tensor&>(transformed_output).tensor<T, 4>(), in operator ()()
|