Searched defs:src_md (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_concat_op.cc | 305 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
|
D | mkl_softmax_op.cc | 121 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
|
D | mkl_relu_op.cc | 44 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
|
D | mkl_fused_batch_norm_op.cc | 147 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
|
D | mkl_lrn_op.cc | 126 memory::desc src_md = src_dnn_shape.GetCurLayout(); in Compute() local
|
D | mkl_conv_grad_filter_ops.cc | 180 std::shared_ptr<mkldnn::memory::desc> src_md; member
|
D | mkl_conv_ops.cc | 186 std::shared_ptr<mkldnn::memory::desc> src_md; member 1030 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
|
D | mkl_pooling_ops_common.h | 125 std::shared_ptr<mkldnn::memory::desc> src_md; member
|