Searched defs:PostOpParam (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| D | Attr.h | 76 struct PostOpParam { struct
78 PostOpParam(float scale, float alpha, float beta, dnnl::algorithm algo, kind_t kind) in PostOpParam() argument
81 PostOpParam(float scale, kind_t kind) : scale_(scale), kind_(kind) {} in PostOpParam() argument
83 PostOpParam( in PostOpParam() argument
95 PostOpParam(int mask, kind_t kind) : mask_(mask), kind_(kind) {} in PostOpParam() argument
98 PostOpParam(at::Tensor& binary, float scale, dnnl::algorithm algo, kind_t kind) in PostOpParam() function
102 float scale_ = 1.0;
104 float alpha_ = 0.0;
105 float beta_ = 0.0;
109 void* binary_ptr_ = nullptr;
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 72 struct PostOpParam { struct
73 string name;
74 std::vector<float> param;
594 struct PostOpParam { struct
595 string name;
596 std::vector<float> param;
597 memory::dims dims;
598 memory::data_type data_type;
599 memory::format_tag format_tag;
|
| D | mkl_quantize_op.cc | 66 struct PostOpParam { struct
67 string name;
68 std::vector<float> param;
|
| D | mkl_conv_ops.cc | 58 struct PostOpParam { struct
59 string name;
60 dnnl::algorithm alg;
61 std::vector<float> param;
62 std::string partial_key;
|