Home
last modified time | relevance | path

Searched refs:compute_data_layout (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc243 se::dnn::DataLayout compute_data_layout; in operator ()() local
246 std::tie(compute_data_layout, filter_layout) = in operator ()()
254 .set_layout(compute_data_layout); in operator ()()
260 .set_layout(compute_data_layout); in operator ()()
Dconv_ops_3d.cc388 se::dnn::DataLayout compute_data_layout; in launch() local
391 std::tie(compute_data_layout, filter_layout) = in launch()
403 .set_layout(compute_data_layout); in launch()
410 .set_layout(compute_data_layout); in launch()
Dconv_grad_ops_3d.cc1292 se::dnn::DataLayout compute_data_layout; in Compute() local
1295 std::tie(compute_data_layout, filter_layout) = in Compute()
1304 .set_layout(compute_data_layout); in Compute()
1311 .set_layout(compute_data_layout); in Compute()
1786 se::dnn::DataLayout compute_data_layout; in Compute() local
1789 std::tie(compute_data_layout, filter_layout) = in Compute()
1801 .set_layout(compute_data_layout); in Compute()
1808 .set_layout(compute_data_layout); in Compute()
Dconv_grad_filter_ops.cc848 se::dnn::DataLayout compute_data_layout; in operator ()() local
851 std::tie(compute_data_layout, filter_layout) = in operator ()()
859 .set_layout(compute_data_layout); in operator ()()
865 .set_layout(compute_data_layout); in operator ()()
Dconv_ops.cc873 se::dnn::DataLayout compute_data_layout; in operator ()() local
876 std::tie(compute_data_layout, filter_layout) = in operator ()()
884 .set_layout(compute_data_layout); in operator ()()
890 .set_layout(compute_data_layout); in operator ()()