Home
last modified time | relevance | path

Searched defs:src_md (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_concat_op.cc305 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
316 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
338 auto src_md = in Compute() local
Dmkl_softmax_op.cc121 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
Dmkl_relu_op.cc44 memory::desc src_md; member in tensorflow::MklEltwiseFwdParams
49 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams()
111 std::shared_ptr<memory::desc> src_md; member
302 std::shared_ptr<memory::desc> src_md; member
472 memory::desc src_md({}, memory::data_undef, memory::format_undef); in Compute() local
587 memory::desc src_md({}, memory::data_undef, memory::format_undef); in Compute() local
Dmkl_fused_batch_norm_op.cc147 auto src_md = memory::desc({fwdParams.src_dims}, MklDnnType<T>(), in Setup() local
388 auto src_md = memory::desc({bwdParams.src_dims}, MklDnnType<T>(), in Setup() local
615 auto src_md = dnn_shape_src.IsMklTensor() in Compute() local
935 memory::desc src_md = in Compute() local
Dmkl_lrn_op.cc126 memory::desc src_md = src_dnn_shape.GetCurLayout(); in Compute() local
Dmkl_conv_grad_filter_ops.cc180 std::shared_ptr<mkldnn::memory::desc> src_md; member
Dmkl_conv_ops.cc186 std::shared_ptr<mkldnn::memory::desc> src_md; member
1030 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
Dmkl_pooling_ops_common.h125 std::shared_ptr<mkldnn::memory::desc> src_md; member