Searched refs:toConstTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 109 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
112 toConstTensor(transformed_output).template tensor<T, 5>(), in Compute()
238 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
241 toConstTensor(transformed_input_backprop).template tensor<T, 5>(), in Compute()
|
| D | pooling_ops_common.cc | 274 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
278 toConstTensor(transformed_output).template tensor<RT, 4>(), in Compute()
427 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
430 toConstTensor(transformed_input_backprop).template tensor<T, 4>(), in Compute()
|
| D | conv_grad_ops_3d.cc | 1445 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
1448 toConstTensor(pre_transformed_in_backprop).template tensor<T, 5>(), in Compute()
1831 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
1834 toConstTensor(pre_transformed_filter_backprop).template tensor<T, 5>(), in Compute()
|
| D | conv_grad_filter_ops.cc | 900 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in operator ()() local
903 toConstTensor(pre_transformed_filter_backprop).template tensor<T, 4>(), in operator ()()
|
| D | conv_grad_input_ops.cc | 1040 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in operator ()() local
1043 toConstTensor(pre_transformed_in_backprop).template tensor<T, 4>(), in operator ()()
|