Searched refs:Tfilter (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 100 template <typename T, typename Tinput, typename Tfilter, typename Tbias,
120 void Execute(const Tinput* src_data, const Tfilter* filter_data, in Execute()
125 static_cast<void*>(const_cast<Tfilter*>(filter_data))); in Execute()
145 void Execute(const Tinput* src_data, const Tfilter* filter_data, in Execute()
150 static_cast<void*>(const_cast<Tfilter*>(filter_data))); in Execute()
220 {convFwdDims.filter_dims}, MklDnnType<Tfilter>(), 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()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 112 DataType Tfilter) { in IsMklOp() argument
117 return (Tinput == DT_QUINT8 && Tfilter == DT_QINT8); in IsMklOp()
|
| D | mkl_layout_pass.cc | 2388 DataType Tinput, Tfilter, out_type; in CopyAttrsQuantizedConv2D() local
2396 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "Tfilter", &Tfilter)); in CopyAttrsQuantizedConv2D()
2410 nb->Attr("Tfilter", Tfilter); in CopyAttrsQuantizedConv2D()
3239 DataType Tinput, Tfilter; in CheckForQuantizedNodeRewrite() local
3241 GetNodeAttr(n->def(), "Tfilter", &Tfilter).ok())) { in CheckForQuantizedNodeRewrite()
3245 Tinput, Tfilter)) { in CheckForQuantizedNodeRewrite()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 46739 type_attr: "Tfilter"
46783 name: "Tfilter"
46834 type_attr: "Tfilter"
46878 name: "Tfilter"
46941 type_attr: "Tfilter"
46985 name: "Tfilter"
47048 type_attr: "Tfilter"
47092 name: "Tfilter"
47155 type_attr: "Tfilter"
47199 name: "Tfilter"
[all …]
|
| D | ops_history.v1.pbtxt | 47322 type_attr: "Tfilter"
47366 name: "Tfilter"
47417 type_attr: "Tfilter"
47461 name: "Tfilter"
47524 type_attr: "Tfilter"
47568 name: "Tfilter"
47631 type_attr: "Tfilter"
47675 name: "Tfilter"
47738 type_attr: "Tfilter"
47782 name: "Tfilter"
[all …]
|
| D | ops_history.v0.pbtxt | 21240 type_attr: "Tfilter"
21284 name: "Tfilter"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 24032 type_attr: "Tfilter"
24076 name: "Tfilter"
24139 type_attr: "Tfilter"
24183 name: "Tfilter"
24254 type_attr: "Tfilter"
24306 name: "Tfilter"
24377 type_attr: "Tfilter"
24429 name: "Tfilter"
24500 type_attr: "Tfilter"
24548 name: "Tfilter"
[all …]
|