Searched refs:ConvertMklToTF (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_quantized_conv_ops_test.cc | 51 void ConvertMklToTF(DataType dtype, const Tensor& input, in ConvertMklToTF() function in tensorflow::ConvMklToTF
195 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
258 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
340 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
394 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
451 conv_comp.ConvertMklToTF<qint32>(DT_QINT32, output, output_mkl_metadata, in TEST_F()
|
| D | mkl_lrn_op.cc | 112 ConvertMklToTF<T>(context, src_tensor, src_dnn_shape); in Compute()
534 input_gradient_tensor = ConvertMklToTF<T>( in MklDefaultToEigen()
541 orig_input_tensor = ConvertMklToTF<T>( in MklDefaultToEigen()
548 orig_output_tensor = ConvertMklToTF<T>( in MklDefaultToEigen()
|
| D | mkl_concat_op.cc | 458 ConvertMklToTF<T>(context, values[i], mkl_input_shapes[i]); in CallEigenVersion()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 803 inline Tensor ConvertMklToTF(OpKernelContext* context, const Tensor& mkl_tensor, in ConvertMklToTF() function
832 inline Tensor ConvertMklToTF(OpKernelContext* context, const Tensor& mkl_tensor, in ConvertMklToTF() function
|