| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 114 typename Toutput>
115 class MklDnnQuantizedMatMulOp : public MklDnnMatMulOpBase<Tweight, Toutput> {
140 : MklDnnMatMulOpBase<Tweight, Toutput>(context) { in MklDnnQuantizedMatMulOp()
223 MklDnnMatMulFwdPrimitive<float, Tinput, Tweight, Tbias, Toutput>* in Compute()
236 Toutput>::Get(matmul_fwd_dims, 0); in Compute()
244 Toutput* dst_data = in Compute()
245 reinterpret_cast<Toutput*>(dst_tensor->flat<Toutput>().data()); in Compute()
308 if (std::is_same<Toutput, quint8>::value || in Compute()
309 std::is_same<Toutput, qint8>::value) { in Compute()
319 if (std::is_same<Toutput, quint8>::value || in Compute()
[all …]
|
| D | mkl_conv_ops.cc | 97 template <typename Tinput, typename Tfilter, typename Tbias, typename Toutput>
115 const Tbias* bias_data, const Toutput* dst_data, in Execute()
128 static_cast<void*>(const_cast<Toutput*>(dst_data)), *fwd_stream); in Execute()
139 static_cast<void*>(const_cast<Toutput*>(dst_data))); in Execute()
163 const Toutput* dst_data, std::shared_ptr<stream> fwd_stream) { in Execute()
225 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), 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_matmul_ops_common.h | 82 typename Toutput>
102 const Tbias* bias_data, Toutput* dst_data, in Execute()
188 MklDnnType<Toutput>(), in Setup()
291 typename Toutput>
294 static MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>* Get( in Get()
296 MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>* matmul_fwd = in Get()
302 new MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>( in Get()
307 MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>*>( in Get()
309 Toutput>::GetInstance() in Get()
313 new MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>( in Get()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_add_op.cc | 44 template <class T, class Toutput>
49 Toutput* output) { in ScalarAddition()
50 const Toutput scalar_in_output_range = RequantizeInNewRange<T, Toutput>( in ScalarAddition()
53 const Toutput full_input_in_output_range = RequantizeInNewRange<T, Toutput>( in ScalarAddition()
158 template <class T, class Toutput>
162 Toutput* output) { in VectorAddition()
164 const Toutput x_in_output_range = RequantizeInNewRange<T, Toutput>( in VectorAddition()
166 const Toutput y_in_output_range = RequantizeInNewRange<T, Toutput>( in VectorAddition()
288 template <class T, class Toutput>
293 float output_min, float output_max, Toutput* output) { in VectorTensorAddition()
[all …]
|
| D | quantized_mul_op.cc | 35 template <class T, class Toutput>
38 int32 scalar_input_offset, Toutput* output) { in ScalarMultiply()
112 template <class T, class Toutput>
115 Toutput* output) { in VectorMultiply()
188 template <class T, class Toutput>
192 Toutput* output) { in VectorTensorMultiply()
279 template <class T, class Toutput>
314 Toutput* z_data = z->flat<Toutput>().data(); in Compute()
319 ScalarMultiply<T, Toutput>(context, y_data, offset_y, y.NumElements(), in Compute()
322 ScalarMultiply<T, Toutput>(context, x_data, offset_x, x.NumElements(), in Compute()
[all …]
|
| D | quantized_matmul_op.cc | 66 template <class T1, class T2, class Toutput>
125 Toutput* c_data = c->flat<Toutput>().data(); in Compute()
136 std::is_same<T2, quint8>() && std::is_same<Toutput, qint32>() && in Compute()
144 std::is_same<Toutput, qint32>() && (offset_c == 0) && in Compute()
171 ReferenceGemm<T1, T2, Toutput>( in Compute()
178 QuantizationRangeForMultiplication<T1, T2, Toutput>( in Compute()
|
| D | depthtospace_op.cc | 113 auto Toutput = outputs_tensor->tensor<T, kDims>(); in Compute() local
126 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
136 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
|
| D | scatter_nd_op_cpu_impl.h | 114 typename TTypes<T, 2>::Tensor Toutput) {
143 auto input_chip = Toutput.template chip<0>(i);
166 typename TTypes<T, 2>::Tensor Toutput)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | QuantizedMul.pbtxt | 29 type_attr: "Toutput"
66 name: "Toutput"
111 type_attr: "Toutput"
148 name: "Toutput"
193 type_attr: "Toutput"
230 name: "Toutput"
|
| D | QuantizedAdd.pbtxt | 29 type_attr: "Toutput"
66 name: "Toutput"
111 type_attr: "Toutput"
148 name: "Toutput"
193 type_attr: "Toutput"
230 name: "Toutput"
|
| D | QuantizedMatMul.pbtxt | 29 type_attr: "Toutput"
66 name: "Toutput"
140 type_attr: "Toutput"
177 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndRelu.pbtxt | 33 type_attr: "Toutput"
70 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndDequantize.pbtxt | 41 type_attr: "Toutput"
80 name: "Toutput"
|
| D | QuantizedMatMulWithBias.pbtxt | 33 type_attr: "Toutput"
80 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt | 41 type_attr: "Toutput"
88 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndRequantize.pbtxt | 41 type_attr: "Toutput"
88 name: "Toutput"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | QuantizedMul.pbtxt | 29 type_attr: "Toutput"
66 name: "Toutput"
111 type_attr: "Toutput"
148 name: "Toutput"
193 type_attr: "Toutput"
230 name: "Toutput"
|
| D | QuantizedAdd.pbtxt | 29 type_attr: "Toutput"
66 name: "Toutput"
111 type_attr: "Toutput"
148 name: "Toutput"
193 type_attr: "Toutput"
230 name: "Toutput"
|
| D | QuantizedMatMul.pbtxt | 29 type_attr: "Toutput"
66 name: "Toutput"
140 type_attr: "Toutput"
177 name: "Toutput"
|
| D | XlaRecvFromHost.pbtxt | 5 type_attr: "Toutput"
8 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndRelu.pbtxt | 33 type_attr: "Toutput"
70 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndDequantize.pbtxt | 41 type_attr: "Toutput"
80 name: "Toutput"
|
| D | QuantizedMatMulWithBias.pbtxt | 33 type_attr: "Toutput"
80 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndRequantize.pbtxt | 41 type_attr: "Toutput"
88 name: "Toutput"
|
| D | QuantizedMatMulWithBiasAndReluAndRequantize.pbtxt | 41 type_attr: "Toutput"
88 name: "Toutput"
|