Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_quantize_op.cc68 PostOpParam post_op_params; member
135 auto const& post_op_params = fwdParams.post_op_params; in Setup() local
138 DCHECK(post_op_params.name == "scale"); in Setup()
139 DCHECK_EQ(post_op_params.param.size(), 1); in Setup()
141 scales.push_back(post_op_params.param[0]); in Setup()
187 if (fwdParams.post_op_params.name == "scale") { in CreateKey()
188 DCHECK_EQ(fwdParams.post_op_params.param.size(), 1); in CreateKey()
189 key_creator.AddAsKey(fwdParams.post_op_params.name); in CreateKey()
190 key_creator.AddAsKey(fwdParams.post_op_params.param[0]); in CreateKey()
470 fwdParams.post_op_params.name = "scale"; in Compute()
[all …]
Dmkl_matmul_op_fused.cc271 params.post_op_params.push_back({"relu", {1.0, 0.0, 0.0}}); in ExtendMklDnnMatMulFwdParams()
273 params.post_op_params.push_back({"relu6", {1.0, 6.0, 0.0}}); in ExtendMklDnnMatMulFwdParams()
275 params.post_op_params.push_back({"elu", {1.0, 1.0, 0.0}}); in ExtendMklDnnMatMulFwdParams()
277 params.post_op_params.push_back({"tanh", {1.0, 0.0, 0.0}}); in ExtendMklDnnMatMulFwdParams()
279 params.post_op_params.push_back({"sum", {1.0}}); in ExtendMklDnnMatMulFwdParams()
281 params.post_op_params.push_back( in ExtendMklDnnMatMulFwdParams()
Dmkl_matmul_ops_common.h59 std::vector<PostOpParam> post_op_params; member
202 auto const& post_op_params = matmul_fwd_params.post_op_params; in Setup() local
205 if (!post_op_params.empty()) { in Setup()
206 for (auto const& post_op_param : post_op_params) { in Setup()
344 for (auto const& post_op_param : mkldnn_matmul_fwd_dims.post_op_params) { in CreateKey()
Dmkl_conv_ops.cc76 std::vector<PostOpParam> post_op_params; member
250 auto const& post_op_params = convFwdDims.post_op_params; in Setup() local
253 if (!post_op_params.empty()) { in Setup()
254 for (auto const& post_op_param : post_op_params) { in Setup()
379 for (auto const& post_op_param : convFwdDims.post_op_params) { in CreateKey()
829 params.post_op_params.push_back( in ExtendConvFwdParams()
833 params.post_op_params.push_back( in ExtendConvFwdParams()
1496 params.post_op_params.push_back({"output_scale", mkldnn::algorithm::undef, in ExtendConvFwdParams()
1671 params.post_op_params.push_back( in ExtendConvFwdParams()
1719 params.post_op_params.push_back({"sum", in ExtendConvFwdParams()
[all …]
Dmkl_qmatmul_op.cc384 params.post_op_params.push_back({"output_scale", output_scale}); in ExtendMklDnnMatMulFwdParams()
527 params.post_op_params.push_back({"relu", {1.0, 0.0, 0.0}}); in ExtendMklDnnMatMulFwdParams()