| /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()
123 static_cast<void*>(const_cast<Tinput*>(src_data))); in Execute()
145 void Execute(const Tinput* src_data, const Tfilter* filter_data, in Execute()
148 static_cast<void*>(const_cast<Tinput*>(src_data))); in Execute()
217 {convFwdDims.src_dims}, MklDnnType<Tinput>(), 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 …]
|
| D | depthtospace_op.cc | 112 auto Tinput = input.tensor<T, kDims>(); in Compute() local
126 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_, in Compute()
137 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
|
| D | spacetodepth_op.cc | 114 auto Tinput = input.tensor<T, kDims>(); in Compute() local
128 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_, in Compute()
139 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_graph_util.h | 111 static inline bool IsMklOp(const string& op_name, DataType Tinput, 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
2395 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "Tinput", &Tinput)); in CopyAttrsQuantizedConv2D()
2409 nb->Attr("Tinput", Tinput); in CopyAttrsQuantizedConv2D()
2431 DataType Tinput, out_type; in CopyAttrsRequantize() local
2434 TF_CHECK_OK(GetNodeAttr(orig_node->def(), "Tinput", &Tinput)); in CopyAttrsRequantize()
2438 nb->Attr("Tinput", Tinput); in CopyAttrsRequantize()
3239 DataType Tinput, Tfilter; in CheckForQuantizedNodeRewrite() local
3240 if (!(GetNodeAttr(n->def(), "Tinput", &Tinput).ok() && in CheckForQuantizedNodeRewrite()
3245 Tinput, Tfilter)) { in CheckForQuantizedNodeRewrite()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Requantize.pbtxt | 40 name: "Tinput"
48 The type of the output. Should be a lower bit depth than Tinput.
|
| D | api_def_QuantizeDownAndShrinkRange.pbtxt | 28 name: "Tinput"
36 The type of the output. Should be a lower bit depth than Tinput.
|
| D | api_def_RequantizationRange.pbtxt | 28 name: "Tinput"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 45690 type_attr: "Tinput"
45713 name: "Tinput"
45743 type_attr: "Tinput"
45766 name: "Tinput"
46317 type_attr: "Tinput"
46329 type_attr: "Tinput"
46341 type_attr: "Tinput"
46353 type_attr: "Tinput"
46365 type_attr: "Tinput"
46388 name: "Tinput"
[all …]
|
| D | ops_history.v1.pbtxt | 46273 type_attr: "Tinput"
46296 name: "Tinput"
46326 type_attr: "Tinput"
46349 name: "Tinput"
46900 type_attr: "Tinput"
46912 type_attr: "Tinput"
46924 type_attr: "Tinput"
46936 type_attr: "Tinput"
46948 type_attr: "Tinput"
46971 name: "Tinput"
[all …]
|
| D | ops_history.v0.pbtxt | 20841 type_attr: "Tinput"
20864 name: "Tinput"
21005 type_attr: "Tinput"
21017 type_attr: "Tinput"
21029 type_attr: "Tinput"
21041 type_attr: "Tinput"
21053 type_attr: "Tinput"
21076 name: "Tinput"
21236 type_attr: "Tinput"
21271 name: "Tinput"
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 23537 type_attr: "Tinput"
23560 name: "Tinput"
23797 type_attr: "Tinput"
23809 type_attr: "Tinput"
23821 type_attr: "Tinput"
23833 type_attr: "Tinput"
23845 type_attr: "Tinput"
23868 name: "Tinput"
24028 type_attr: "Tinput"
24063 name: "Tinput"
[all …]
|