Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_input_ops.cc239 se::dnn::DataLayout compute_data_layout; in operator ()() local
242 std::tie(compute_data_layout, filter_layout) = in operator ()()
250 .set_layout(compute_data_layout); in operator ()()
256 .set_layout(compute_data_layout); in operator ()()
H A Dconv_ops_fused_impl.h487 se::dnn::DataLayout compute_data_layout;
489 std::tie(compute_data_layout, filter_layout) =
497 .set_layout(compute_data_layout);
509 .set_layout(compute_data_layout);
523 .set_layout(compute_data_layout);
H A Dconv_ops_3d.cc396 se::dnn::DataLayout compute_data_layout; in launch() local
399 std::tie(compute_data_layout, filter_layout) = in launch()
411 .set_layout(compute_data_layout); in launch()
418 .set_layout(compute_data_layout); in launch()
H A Dconv_grad_ops_3d.cc1394 se::dnn::DataLayout compute_data_layout; in Compute() local
1397 std::tie(compute_data_layout, filter_layout) = in Compute()
1406 .set_layout(compute_data_layout); in Compute()
1413 .set_layout(compute_data_layout); in Compute()
1790 se::dnn::DataLayout compute_data_layout; in Compute() local
1793 std::tie(compute_data_layout, filter_layout) = in Compute()
1805 .set_layout(compute_data_layout); in Compute()
1812 .set_layout(compute_data_layout); in Compute()
H A Dconv_grad_filter_ops.cc858 se::dnn::DataLayout compute_data_layout; in operator ()() local
861 std::tie(compute_data_layout, filter_layout) = in operator ()()
869 .set_layout(compute_data_layout); in operator ()()
875 .set_layout(compute_data_layout); in operator ()()
H A Dconv_ops.cc1024 se::dnn::DataLayout compute_data_layout; in operator ()() local
1027 std::tie(compute_data_layout, filter_layout) = in operator ()()
1035 .set_layout(compute_data_layout); in operator ()()
1041 .set_layout(compute_data_layout); in operator ()()