Searched refs:Tbias (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 100 template <typename T, typename Tinput, typename Tfilter, typename Tbias,
121 const Tbias* bias_data, const Toutput* dst_data) { in Execute()
127 static_cast<void*>(const_cast<Tbias*>(bias_data))); in Execute()
227 {convFwdDims.bias_dims}, MklDnnType<Tbias>(), memory::format::any)); in Setup()
319 template <typename T, typename Tinput, typename Tfilter, typename Tbias,
323 static MklConvFwdPrimitive<T, Tinput, Tfilter, Tbias, Toutput>* Get( in Get()
325 MklConvFwdPrimitive<T, Tinput, Tfilter, Tbias, Toutput>* conv_fwd = nullptr; in Get()
329 conv_fwd = new MklConvFwdPrimitive<T, Tinput, Tfilter, Tbias, Toutput>( in Get()
334 MklConvFwdPrimitive<T, Tinput, Tfilter, Tbias, Toutput>*>( in Get()
335 MklConvFwdPrimitiveFactory<T, Tinput, Tfilter, Tbias, in Get()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 2423 DataType Tbias; in CopyAttrsQuantizedConv2D() local
2424 Status bias_status = GetNodeAttr(orig_node->def(), "Tbias", &Tbias); in CopyAttrsQuantizedConv2D()
2425 if (bias_status.ToString() == "OK") nb->Attr("Tbias", Tbias); in CopyAttrsQuantizedConv2D()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 48210 type_attr: "Tbias"
48275 name: "Tbias"
48339 type_attr: "Tbias"
48404 name: "Tbias"
48476 type_attr: "Tbias"
48541 name: "Tbias"
48605 type_attr: "Tbias"
48670 name: "Tbias"
48742 type_attr: "Tbias"
48819 name: "Tbias"
[all …]
|
| D | ops_history.v1.pbtxt | 48793 type_attr: "Tbias"
48858 name: "Tbias"
48922 type_attr: "Tbias"
48987 name: "Tbias"
49059 type_attr: "Tbias"
49124 name: "Tbias"
49188 type_attr: "Tbias"
49253 name: "Tbias"
49325 type_attr: "Tbias"
49402 name: "Tbias"
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 24742 type_attr: "Tbias"
24807 name: "Tbias"
24879 type_attr: "Tbias"
24944 name: "Tbias"
25016 type_attr: "Tbias"
25093 name: "Tbias"
25301 type_attr: "Tbias"
25378 name: "Tbias"
|