Home
last modified time | relevance | path

Searched refs:SetElemType (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_input_conversion_op.cc126 mkl_output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
224 mkl_output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
Dmkl_softmax_op.cc147 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
Dmkl_reshape_op.cc189 mkl_shape_output.SetElemType(MklDnnType<T>()); in Compute()
Dmkl_conv_grad_input_ops.cc427 diff_src_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
547 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
Dmkl_aggregate_ops.cc221 output_mkl_shape.SetElemType(MklDnnType<T>()); in Compute()
Dmkl_lrn_op.cc207 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
457 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
Dmkl_pooling_ops_common.h651 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
703 output_mkl_shape.SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
Dmkl_slice_op.cc461 output_mkl_shape->SetElemType(MklDnnType<T>()); in AllocateOutputTensor()
Dmkl_fused_batch_norm_op.cc664 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()
997 dnn_shape_diff_src.SetElemType(MklDnnType<T>()); in Compute()
Dmkl_relu_op.cc511 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()
673 dnn_shape_diff_src.SetElemType(MklDnnType<T>()); in Compute()
Dmkl_concat_op.cc401 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()
Dmkl_conv_ops.cc1267 output_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
1339 filter_mkl_shape.SetElemType(MklDnnType<Tfilter>()); in AllocateFilterOutputTensor()
1779 summand_mkl_shape.SetElemType(MklDnnType<Toutput>()); in AllocateOutputTensor()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h611 inline void SetElemType(memory::data_type dt) { data_.T_ = dt; } in SetElemType() function