Searched refs:SetElemType (Results 1 – 16 of 16) sorted by relevance
125 mkl_output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()226 mkl_output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
71 mkl_shape.SetElemType(MklDnnType<uint8>()); in CreateMklInput()
111 output_mkl_shape.SetElemType(MklDnnType<float>()); in Compute()
446 diff_src_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()573 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
224 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
287 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
202 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()464 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
654 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()706 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
410 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
487 output_mkl_shape->SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
502 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()713 dnn_shape_diff_src.SetElemType(MklDnnType<T>()); in Compute()
865 output_mkl_shape->SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()1020 filter_mkl_shape.SetElemType(MklDnnType<Tfilter>()); in AllocateFilterOutputTensor()1763 summand_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
851 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()1249 dnn_shape_diff_src.SetElemType(MklDnnType<T>()); in Compute()
400 output_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
726 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()
422 inline void SetElemType(memory::data_type dt) { data_.T_ = dt; } in SetElemType() function