Searched refs:toConstTensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 124 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
127 toConstTensor(transformed_output).template tensor<T, 5>(), in Compute()
269 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
272 toConstTensor(transformed_input_backprop).template tensor<T, 5>(), in Compute()
|
| D | pooling_ops_common.cc | 412 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
416 toConstTensor(transformed_output).template tensor<RT, 4>(), in Compute()
749 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
752 toConstTensor(transformed_input_backprop).template tensor<T, 4>(), in Compute()
|
| D | lrn_op.cc | 288 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in launch()
291 toConstTensor(transformed_output).template tensor<T, 4>(), in launch()
618 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in launch()
621 toConstTensor(transformed_output).template tensor<T, 4>(), in launch()
|
| D | conv_grad_input_ops.cc | 542 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in operator ()() local
545 toConstTensor(pre_transformed_in_backprop).template tensor<T, 4>(), in operator ()()
|
| D | conv_grad_ops_3d.cc | 1556 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
1559 toConstTensor(pre_transformed_in_backprop).template tensor<T, 5>(), in Compute()
2025 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in Compute() local
2028 toConstTensor(pre_transformed_filter_backprop).template tensor<T, 5>(), in Compute()
|
| D | conv_grad_filter_ops.cc | 1113 auto toConstTensor = [](const Tensor& x) -> const Tensor { return x; }; in operator ()() local
1116 toConstTensor(pre_transformed_filter_backprop).template tensor<T, 4>(), in operator ()()
|