Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tests/
Dpooling_ops_test.py548 gen_nn_ops.max_pool_grad,
573 gen_nn_ops.max_pool_grad,
582 gen_nn_ops.max_pool_grad,
590 gen_nn_ops.max_pool_grad,
/external/tensorflow/tensorflow/python/kernel_tests/
Dfractional_max_pool_op_test.py381 input_backprop_tensor = gen_nn_ops.max_pool_grad(
416 input_backprop_tensor = gen_nn_ops.max_pool_grad(
Dpooling_ops_test.py959 out_op = gen_nn_ops.max_pool_grad(t, orig_out, grad_in, ksize, strides,
1510 pool_func = gen_nn_ops.max_pool_grad_v2 if v2 else gen_nn_ops.max_pool_grad
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc286 csinfo_.max_pool_grad = "MaxPoolGrad"; in MklLayoutRewritePass()
524 rinfo_.push_back({csinfo_.max_pool_grad, in MklLayoutRewritePass()
525 mkl_op_registry::GetMklOpName(csinfo_.max_pool_grad), in MklLayoutRewritePass()
702 wsinfo_.push_back({csinfo_.max_pool, csinfo_.max_pool_grad, 0, 1, 1, 3}); in MklLayoutRewritePass()
945 string max_pool_grad; member
1712 if (e->dst()->type_string() == csinfo_.max_pool_grad && in MaxpoolGradRewrite()
3733 n->type_string() == csinfo_.max_pool_grad || in CheckForNodeRewrite()
/external/tensorflow/tensorflow/python/ops/
Dnn_grad.py684 return gen_nn_ops.max_pool_grad(
762 gen_nn_ops.max_pool_grad(
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py1152 max_pool_grad = gen_nn_ops.max_pool_grad_v2(conv, conv, conv, ksize,
1154 output = array_ops.identity(max_pool_grad)