Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_eltwise_activation_base_op.h56 memory::desc src_md; variable
61 MklEltwiseFwdActivationParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdActivationParams()
126 std::shared_ptr<memory::desc> src_md; member
H A Dmkl_fused_batch_norm_op.cc54 memory::desc src_md; member
207 auto src_md = fwdParams.src_md; in Setup() local
398 memory::desc src_md; member
545 auto src_md = bwdParams.src_md; in Setup() local
833 auto src_md = dnn_shape_src.IsMklTensor() in Compute() local
1263 memory::desc src_md = in Compute() local
H A Dmkl_relu_op.cc48 memory::desc src_md; member in tensorflow::MklEltwiseFwdParams
53 MklEltwiseFwdParams(memory::dims src_dims, memory::desc src_md, in MklEltwiseFwdParams()
117 std::shared_ptr<memory::desc> src_md; member
304 std::shared_ptr<memory::desc> src_md; member
465 memory::desc src_md({}, memory::data_type::undef, in Compute() local
623 memory::desc src_md({}, memory::data_type::undef, in Compute() local
H A Dmkl_concat_op.cc331 std::vector<dnnl::memory::desc> src_md; member
622 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
633 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
661 auto src_md = in Compute() local
H A Dmkl_pooling_ops_common.h52 memory::desc src_md; member
128 std::shared_ptr<dnnl::memory::desc> src_md; member
269 std::shared_ptr<dnnl::memory::desc> src_md; member
H A Dmkl_quantize_op.cc63 memory::desc src_md; member
379 auto src_md = in Compute() local
H A Dmkl_dequantize_op.cc105 auto src_md = memory::desc(src_dims, MklDnnType<T>(), dst_layout_type); in Compute() local
H A Dmkl_layer_norm_op.cc67 auto src_md = in Compute() local
H A Dmkl_matmul_op_fused.cc226 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
H A Dmkl_avgpooling_op.cc237 memory::desc src_md = in Compute() local
H A Dmkl_softmax_op.cc103 std::shared_ptr<memory::desc> src_md; member
H A Dmkl_maxpooling_op.cc290 memory::desc src_md = in Compute() local
H A Dmkl_qmatmul_op.cc210 auto src_md = in Compute() local
H A Dmkl_lrn_op.cc130 memory::desc src_md = src_dnn_shape.GetCurLayout(); in Compute() local
H A Dmkl_conv_grad_filter_ops.cc170 std::shared_ptr<dnnl::memory::desc> src_md; member
H A Dmkl_conv_ops.cc235 std::shared_ptr<dnnl::memory::desc> src_md; member
748 auto src_md = in Compute() local
H A Dmkl_matmul_ops_common.h182 std::shared_ptr<dnnl::memory::desc> src_md; member