Searched refs:ConvertMklToTF (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_quantized_conv_ops_test.cc | 49 void ConvertMklToTF(DataType dtype, const Tensor& input, in ConvertMklToTF() function in tensorflow::ConvMklToTF
176 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in RunQuantizedDepthwiseConv2DOp()
277 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
381 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
443 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
525 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
579 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
636 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
|
| D | mkl_lrn_op.cc | 115 ConvertMklToTF<T>(context, src_tensor, src_dnn_shape, in Compute()
516 ConvertMklToTF<T>(context, MklGetInput(context, kIdxGradient), in MklDefaultToEigen()
523 OP_REQUIRES_OK(context, ConvertMklToTF<T>( in MklDefaultToEigen()
531 OP_REQUIRES_OK(context, ConvertMklToTF<T>( in MklDefaultToEigen()
|
| D | mkl_concat_op.cc | 830 context, ConvertMklToTF<T>(context, values[i], mkl_input_shapes[i], in CallEigenVersion()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 623 inline Status ConvertMklToTF(OpKernelContext* context, in ConvertMklToTF() function
|