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"
|