HomeSort by relevance Sort by last modified time
    Searched refs:diff_src_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc 485 Tensor* diff_src_tensor = nullptr; local
595 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor,
600 diff_src.SetUsrMem(common_md, diff_src_tensor);
674 Tensor* diff_src_tensor = nullptr; local
681 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor,
683 void* out_o = static_cast<void*>(diff_src_tensor->flat<T>().data());
741 Tensor* diff_src_tensor = nullptr; local
748 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor,
750 void* out_o = static_cast<void*>(diff_src_tensor->flat<T>().data());
812 Tensor* diff_src_tensor = nullptr local
    [all...]
mkl_fused_batch_norm_op.cc 1105 Tensor* diff_src_tensor = nullptr; variable
    [all...]

Completed in 601 milliseconds