Searched refs:variance_mem (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 96 context_.variance_mem->set_data_handle(static_cast<void*>(variance_data), in Execute()
114 context_.variance_mem->set_data_handle(static_cast<void*>(variance_data)); in Execute()
133 context_.variance_mem->set_data_handle(DummyData); in Execute()
161 std::shared_ptr<mkldnn::memory> variance_mem; member
180 variance_mem(nullptr), in BatchNormFwdContext()
224 context_.variance_mem.reset( in Setup()
253 {MKLDNN_ARG_VARIANCE, *context_.variance_mem}, in Setup()
261 {MKLDNN_ARG_VARIANCE, *context_.variance_mem}, in Setup()
270 {MKLDNN_ARG_VARIANCE, *context_.variance_mem}, in Setup()
277 {MKLDNN_ARG_VARIANCE, *context_.variance_mem}, in Setup()
[all …]
|