HomeSort by relevance Sort by last modified time
    Searched defs:diff_src_mem (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_input_ops.cc 100 context_.diff_src_mem->set_data_handle(
110 context_.diff_src_mem->set_data_handle(DummyData);
135 std::shared_ptr<mkldnn::memory> diff_src_mem; member in struct:tensorflow::MklConvBwdInputPrimitive::ConvBwdInputContext
161 diff_src_mem(nullptr),
207 context_.diff_src_mem.reset(new memory(
229 *context_.diff_src_mem));
mkl_pooling_ops_common.h 264 std::shared_ptr<mkldnn::memory> diff_src_mem; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
288 diff_src_mem(nullptr),
mkl_relu_op.cc 269 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data));
275 context_.diff_src_mem->set_data_handle(DummyData);
296 std::shared_ptr<memory> diff_src_mem; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
326 diff_src_mem(nullptr),
365 context_.diff_src_mem.reset(new memory(
371 *context_.diff_src_mem));
    [all...]
mkl_fused_batch_norm_op.cc 325 context_.diff_src_mem->set_data_handle(static_cast<void*>(diff_src_data));
338 context_.diff_src_mem->set_data_handle(DummyData);
350 return (*context_.diff_src_mem).get_primitive_desc();
365 std::shared_ptr<mkldnn::memory> diff_src_mem; member in struct:tensorflow::MklFusedBatchNormBwdPrimitive::BatchNormBwdContext
379 diff_src_mem(nullptr),
433 context_.diff_src_mem.reset(new memory({src_md, cpu_engine_}, DummyData));
438 *context_.diff_src_mem, *context_.diff_weights_mem));
    [all...]

Completed in 58 milliseconds