Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_qmatmul_op.cc113 template <typename Device, typename Tinput, typename Tweight, typename Tbias,
223 MklDnnMatMulFwdPrimitive<float, Tinput, Tweight, Tbias, Toutput>* in Compute()
235 MklDnnMatMulFwdPrimitiveFactory<float, Tinput, Tweight, Tbias, in Compute()
294 Tbias* bias_data = this->GetBiasHandle( in Compute()
354 params.dtypes.append(typeid(Tbias).name()); in ExtendMklDnnMatMulFwdParams()
393 Tbias* GetBiasHandle( in GetBiasHandle()
401 if (std::is_same<Tbias, qint32>::value) { in GetBiasHandle()
402 return static_cast<Tbias*>( in GetBiasHandle()
403 const_cast<Tbias*>(bias_tensor.flat<Tbias>().data())); in GetBiasHandle()
463 return reinterpret_cast<Tbias*>(comp_bias_); in GetBiasHandle()
[all …]
Dmkl_conv_ops.cc97 template <typename Tinput, typename Tfilter, typename Tbias, typename Toutput>
115 const Tbias* bias_data, const Toutput* dst_data, in Execute()
125 static_cast<void*>(const_cast<Tbias*>(bias_data)), *fwd_stream); in Execute()
136 static_cast<void*>(const_cast<Tbias*>(bias_data))); in Execute()
229 MklDnnType<Tbias>(), in Setup()
296 {{convFwdDims.bias_dims}, MklDnnType<Tbias>(), memory::format_tag::x}, in Setup()
320 template <typename Tinput, typename Tfilter, typename Tbias, typename Toutput>
323 static MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>* Get( in Get()
325 MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>* conv_fwd = nullptr; in Get()
330 new MklConvFwdPrimitive<Tinput, Tfilter, Tbias, Toutput>(convFwdDims); in Get()
[all …]
Dmkl_matmul_ops_common.h81 template <typename T, typename Tinput, typename Tweight, typename Tbias,
102 const Tbias* bias_data, Toutput* dst_data, in Execute()
110 static_cast<void*>(const_cast<Tbias*>(bias_data))); in Execute()
119 static_cast<void*>(const_cast<Tbias*>(bias_data))); in Execute()
192 MklDnnType<Tbias>(), in Setup()
272 MklDnnType<Tbias>(), in Setup()
290 template <typename T, typename Tinput, typename Tweight, typename Tbias,
294 static MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>* Get( in Get()
296 MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>* matmul_fwd = in Get()
302 new MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>( in Get()
[all …]
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DQuantizedConv2DWithBiasAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
142 type_attr: "Tbias"
207 name: "Tbias"
DQuantizedConv2DWithBiasAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
142 type_attr: "Tbias"
207 name: "Tbias"
DQuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
142 type_attr: "Tbias"
207 name: "Tbias"
DQuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
90 name: "Tbias"
167 type_attr: "Tbias"
244 name: "Tbias"
DQuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
90 name: "Tbias"
167 type_attr: "Tbias"
244 name: "Tbias"
DQuantizedMatMulWithBiasAndDequantize.pbtxt13 type_attr: "Tbias"
70 name: "Tbias"
DQuantizedMatMulWithBias.pbtxt13 type_attr: "Tbias"
70 name: "Tbias"
DQuantizedMatMulWithBiasAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
DQuantizedMatMulWithBiasAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DQuantizedConv2DWithBiasAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
142 type_attr: "Tbias"
207 name: "Tbias"
DQuantizedConv2DWithBiasAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
142 type_attr: "Tbias"
207 name: "Tbias"
DQuantizedConv2DWithBiasSignedSumAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
90 name: "Tbias"
167 type_attr: "Tbias"
244 name: "Tbias"
DQuantizedConv2DWithBiasSumAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
90 name: "Tbias"
167 type_attr: "Tbias"
244 name: "Tbias"
DQuantizedMatMulWithBiasAndDequantize.pbtxt13 type_attr: "Tbias"
70 name: "Tbias"
DQuantizedMatMulWithBias.pbtxt13 type_attr: "Tbias"
70 name: "Tbias"
DQuantizedMatMulWithBiasAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
DQuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
DQuantizedMatMulWithBiasAndReluAndRequantize.pbtxt13 type_attr: "Tbias"
78 name: "Tbias"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt77 name: "Tbias"
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc2778 DataType Tbias; in CopyAttrsQuantizedConv2D() local
2779 Status bias_status = GetNodeAttr(orig_node->def(), "Tbias", &Tbias); in CopyAttrsQuantizedConv2D()
2780 if (bias_status.ToString() == "OK") nb->Attr("Tbias", Tbias); in CopyAttrsQuantizedConv2D()
2799 DataType Tbias; in CopyAttrsQuantizedMatMulWithBiasAndDequantize() local
2800 Status bias_status = GetNodeAttr(orig_node->def(), "Tbias", &Tbias); in CopyAttrsQuantizedMatMulWithBiasAndDequantize()
2801 if (bias_status.ToString() == "OK") nb->Attr("Tbias", Tbias); in CopyAttrsQuantizedMatMulWithBiasAndDequantize()
2824 DataType Tbias; in CopyAttrsQuantizedMatMulWithBias() local
2825 Status bias_status = GetNodeAttr(orig_node->def(), "Tbias", &Tbias); in CopyAttrsQuantizedMatMulWithBias()
2826 if (bias_status.ToString() == "OK") nb->Attr("Tbias", Tbias); in CopyAttrsQuantizedMatMulWithBias()
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt31726 type_attr: "Tbias"
31791 name: "Tbias"
31863 type_attr: "Tbias"
31928 name: "Tbias"
32000 type_attr: "Tbias"
32077 name: "Tbias"
32285 type_attr: "Tbias"
32362 name: "Tbias"
32784 type_attr: "Tbias"
32849 name: "Tbias"
[all …]
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt47547 type_attr: "Tbias"
47612 name: "Tbias"
47676 type_attr: "Tbias"
47741 name: "Tbias"
47813 type_attr: "Tbias"
47878 name: "Tbias"
47942 type_attr: "Tbias"
48007 name: "Tbias"
48079 type_attr: "Tbias"
48156 name: "Tbias"
[all …]