Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_relu_op.cc45 memory::desc src_md; member in tensorflow::MklEltwiseFwdParams
50 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams()
111 std::shared_ptr<memory::desc> src_md; member
297 std::shared_ptr<memory::desc> src_md; member
464 memory::desc src_md({}, memory::data_type::undef, in Compute() local
622 memory::desc src_md({}, memory::data_type::undef, in Compute() local
Dmkl_fused_batch_norm_op.cc50 memory::desc src_md; member
200 auto src_md = fwdParams.src_md; in Setup() local
388 memory::desc src_md; member
532 auto src_md = bwdParams.src_md; in Setup() local
779 auto src_md = dnn_shape_src.IsMklTensor() in Compute() local
1180 memory::desc src_md = in Compute() local
Dmkl_concat_op.cc325 std::vector<mkldnn::memory::desc> src_md; member
612 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
623 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
651 auto src_md = in Compute() local
Dmkl_dequantize_op.cc85 auto src_md = in Compute() local
Dmkl_quantize_op.cc61 memory::desc src_md; member
370 auto src_md = in Compute() local
Dmkl_matmul_op_fused.cc212 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
Dmkl_pooling_ops_common.h50 memory::desc src_md; member
126 std::shared_ptr<mkldnn::memory::desc> src_md; member
263 std::shared_ptr<mkldnn::memory::desc> src_md; member
Dmkl_avgpooling_op.cc228 memory::desc src_md = in Compute() local
Dmkl_softmax_op.cc95 std::shared_ptr<memory::desc> src_md; member
Dmkl_maxpooling_op.cc285 memory::desc src_md = in Compute() local
Dmkl_qmatmul_op.cc210 auto src_md = in Compute() local
Dmkl_lrn_op.cc130 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_matmul_ops_common.h156 std::shared_ptr<mkldnn::memory::desc> src_md; member
Dmkl_conv_ops.cc232 std::shared_ptr<mkldnn::memory::desc> src_md; member
726 auto src_md = in Compute() local