Searched defs:src_md (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 64 #define IS_SRC_REORDER_NEEDED(src_md, op_pd, op) src_md != op_pd->src_desc() argument
111 #define REORDER_PD_CONSTRUCTOR(src_md, dst_md, engine) \ argument
113 #define REORDER_PD_CONSTRUCTOR_WITH_ATTR(src_md, dst_md, engine, prim_attr) \ argument
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_dequantize_op.cc | 84 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
|
| D | mkl_relu_op.cc | 45 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
621 memory::desc src_md({}, memory::data_type::undef, in Compute() local
|
| D | mkl_fused_batch_norm_op.cc | 49 memory::desc src_md; member
197 auto src_md = fwdParams.src_md; in Setup() local
384 memory::desc src_md; member
526 auto src_md = bwdParams.src_md; in Setup() local
772 auto src_md = dnn_shape_src.IsMklTensor() in Compute() local
1172 memory::desc src_md = in Compute() local
|
| D | mkl_concat_op.cc | 324 std::vector<mkldnn::memory::desc> src_md; member
610 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
621 auto src_md = mkl_input_shapes[k].GetMklLayout(); in Compute() local
649 auto src_md = in Compute() local
|
| D | mkl_quantize_op.cc | 61 memory::desc src_md; member
370 auto src_md = in Compute() local
|
| D | mkl_matmul_op_fused.cc | 212 auto src_md = src_mkl_shape.IsMklTensor() in Compute() local
|
| D | mkl_pooling_ops_common.h | 50 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
|
| D | mkl_avgpooling_op.cc | 227 memory::desc src_md = in Compute() local
|
| D | mkl_softmax_op.cc | 95 std::shared_ptr<memory::desc> src_md; member
|
| D | mkl_maxpooling_op.cc | 284 memory::desc src_md = in Compute() local
|
| D | mkl_qmatmul_op.cc | 209 auto src_md = in Compute() local
|
| D | mkl_lrn_op.cc | 130 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_matmul_ops_common.h | 151 std::shared_ptr<mkldnn::memory::desc> src_md; member
|
| D | mkl_conv_ops.cc | 184 std::shared_ptr<mkldnn::memory::desc> src_md; member
611 auto src_md = in Compute() local
|