Searched refs:Tinput (Results 1 – 25 of 66) sorted by relevance
123
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | QuantizedBatchNormWithGlobalNormalization.pbtxt | 5 type_attr: "Tinput"
17 type_attr: "Tinput"
29 type_attr: "Tinput"
41 type_attr: "Tinput"
53 type_attr: "Tinput"
76 name: "Tinput"
114 type_attr: "Tinput"
126 type_attr: "Tinput"
138 type_attr: "Tinput"
150 type_attr: "Tinput"
[all …]
|
| D | RequantizationRange.pbtxt | 5 type_attr: "Tinput"
24 name: "Tinput"
41 type_attr: "Tinput"
60 name: "Tinput"
|
| D | QuantizeDownAndShrinkRange.pbtxt | 5 type_attr: "Tinput"
28 name: "Tinput"
58 type_attr: "Tinput"
81 name: "Tinput"
|
| D | QuantizedRelu.pbtxt | 5 type_attr: "Tinput"
28 name: "Tinput"
61 type_attr: "Tinput"
84 name: "Tinput"
|
| D | QuantizedRelu6.pbtxt | 5 type_attr: "Tinput"
28 name: "Tinput"
61 type_attr: "Tinput"
84 name: "Tinput"
|
| D | QuantizedReluX.pbtxt | 5 type_attr: "Tinput"
32 name: "Tinput"
65 type_attr: "Tinput"
92 name: "Tinput"
|
| D | Requantize.pbtxt | 5 type_attr: "Tinput"
36 name: "Tinput"
66 type_attr: "Tinput"
97 name: "Tinput"
|
| D | QuantizedConv2D.pbtxt | 5 type_attr: "Tinput"
40 name: "Tinput"
100 type_attr: "Tinput"
135 name: "Tinput"
207 type_attr: "Tinput"
242 name: "Tinput"
|
| D | XlaSendToHost.pbtxt | 5 type_attr: "Tinput"
8 name: "Tinput"
|
| D | QuantizedConv2DAndRelu.pbtxt | 5 type_attr: "Tinput"
40 name: "Tinput"
112 type_attr: "Tinput"
147 name: "Tinput"
|
| D | QuantizedConv2DAndRequantize.pbtxt | 5 type_attr: "Tinput"
48 name: "Tinput"
120 type_attr: "Tinput"
163 name: "Tinput"
|
| D | QuantizedConv2DWithBias.pbtxt | 5 type_attr: "Tinput"
44 name: "Tinput"
116 type_attr: "Tinput"
155 name: "Tinput"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | QuantizedBatchNormWithGlobalNormalization.pbtxt | 5 type_attr: "Tinput"
17 type_attr: "Tinput"
29 type_attr: "Tinput"
41 type_attr: "Tinput"
53 type_attr: "Tinput"
76 name: "Tinput"
114 type_attr: "Tinput"
126 type_attr: "Tinput"
138 type_attr: "Tinput"
150 type_attr: "Tinput"
[all …]
|
| D | RequantizationRange.pbtxt | 5 type_attr: "Tinput"
24 name: "Tinput"
41 type_attr: "Tinput"
60 name: "Tinput"
|
| D | QuantizeDownAndShrinkRange.pbtxt | 5 type_attr: "Tinput"
28 name: "Tinput"
58 type_attr: "Tinput"
81 name: "Tinput"
|
| D | QuantizedRelu.pbtxt | 5 type_attr: "Tinput"
28 name: "Tinput"
61 type_attr: "Tinput"
84 name: "Tinput"
|
| D | QuantizedRelu6.pbtxt | 5 type_attr: "Tinput"
28 name: "Tinput"
61 type_attr: "Tinput"
84 name: "Tinput"
|
| D | QuantizedReluX.pbtxt | 5 type_attr: "Tinput"
32 name: "Tinput"
65 type_attr: "Tinput"
92 name: "Tinput"
|
| D | Requantize.pbtxt | 5 type_attr: "Tinput"
36 name: "Tinput"
66 type_attr: "Tinput"
97 name: "Tinput"
|
| D | QuantizedConv2D.pbtxt | 5 type_attr: "Tinput"
40 name: "Tinput"
100 type_attr: "Tinput"
135 name: "Tinput"
207 type_attr: "Tinput"
242 name: "Tinput"
|
| D | QuantizedConv2DAndRelu.pbtxt | 5 type_attr: "Tinput"
40 name: "Tinput"
112 type_attr: "Tinput"
147 name: "Tinput"
|
| D | QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt | 5 type_attr: "Tinput"
44 name: "Tinput"
116 type_attr: "Tinput"
155 name: "Tinput"
|
| D | QuantizedConv2DWithBias.pbtxt | 5 type_attr: "Tinput"
44 name: "Tinput"
116 type_attr: "Tinput"
155 name: "Tinput"
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 97 template <typename Tinput, typename Tfilter, typename Tbias, typename Toutput>
114 void Execute(const Tinput* src_data, const Tfilter* filter_data, in Execute()
120 static_cast<void*>(const_cast<Tinput*>(src_data)), *fwd_stream); in Execute()
131 static_cast<void*>(const_cast<Tinput*>(src_data))); in Execute()
162 void Execute(const Tinput* src_data, const Tfilter* filter_data, in Execute()
218 {convFwdDims.src_dims}, MklDnnType<Tinput>(), user_data_fmt)); 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 …]
|
| D | mkl_qmatmul_op.cc | 113 template <typename Device, typename Tinput, typename Tweight, typename Tbias,
174 MklDnnData<Tinput> src(&(this->cpu_engine_)); in Compute()
212 : memory::desc(src_dims, MklDnnType<Tinput>(), input_output_fmt); in Compute()
223 MklDnnMatMulFwdPrimitive<float, Tinput, Tweight, Tbias, Toutput>* in Compute()
235 MklDnnMatMulFwdPrimitiveFactory<float, Tinput, Tweight, Tbias, in Compute()
248 Tinput* src_data = nullptr; in Compute()
253 src_data = static_cast<Tinput*>(src.GetOpMem().get_data_handle()); in Compute()
255 src_data = static_cast<Tinput*>( in Compute()
256 const_cast<Tinput*>(src_tensor.flat<Tinput>().data())); in Compute()
352 params.dtypes.append(typeid(Tinput).name()); in ExtendMklDnnMatMulFwdParams()
[all …]
|
123