Searched refs:mean_mem (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 94 context_.mean_mem->set_data_handle(static_cast<void*>(mean_data), in Execute()
113 context_.mean_mem->set_data_handle(static_cast<void*>(mean_data)); in Execute()
132 context_.mean_mem->set_data_handle(DummyData); in Execute()
160 std::shared_ptr<mkldnn::memory> mean_mem; member
179 mean_mem(nullptr), in BatchNormFwdContext()
220 context_.mean_mem.reset( in Setup()
252 {MKLDNN_ARG_MEAN, *context_.mean_mem}, in Setup()
260 {MKLDNN_ARG_MEAN, *context_.mean_mem}, in Setup()
269 {MKLDNN_ARG_MEAN, *context_.mean_mem}, in Setup()
276 {MKLDNN_ARG_MEAN, *context_.mean_mem}, in Setup()
[all …]
|