Home
last modified time | relevance | path

Searched refs:transform_filter (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc282 const auto transform_filter = [&](FilterTensorFormat dst_format) -> Status { in operator ()() local
304 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OIHW)); in operator ()()
306 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OHWI)); in operator ()()
Dconv_ops.cc908 const auto transform_filter = [&](FilterTensorFormat dst_format) -> Status { in operator ()() local
930 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OIHW)); in operator ()()
932 OP_REQUIRES_OK(ctx, transform_filter(FORMAT_OHWI)); in operator ()()