Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.cc80 std::vector<PostOpParam> post_op_params; member
247 auto const& post_op_params = convFwdDims.post_op_params; in Setup() local
250 if (!post_op_params.empty()) { in Setup()
251 for (auto const& post_op_param : post_op_params) { in Setup()
376 for (auto const& post_op_param : convFwdDims.post_op_params) { in CreateKey()
1236 if (fuse_relu_) params.post_op_params.push_back({"relu", {1.0, 0.0, 0.0}}); in ExtendConvFwdParams()
1620 params.post_op_params.push_back({"output_scale", output_scale}); in ExtendConvFwdParams()
1691 params.post_op_params.push_back({"relu", {1.0, 0.0, 0.0}}); in ExtendConvFwdParams()
1744 params.post_op_params.push_back( in ExtendConvFwdParams()
1747 params.post_op_params.push_back( in ExtendConvFwdParams()
[all …]