Searched refs:AllocatePersistentTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 974 void AllocatePersistentTensor(OpKernelContext* context, in AllocatePersistentTensor() function in tensorflow::MklConvOp
1003 void AllocatePersistentTensor(OpKernelContext* context, in AllocatePersistentTensor() function in tensorflow::MklConvOp
1006 AllocatePersistentTensor(context, conv_prim_desc, filter_tensor, nullptr); in AllocatePersistentTensor()
1111 AllocatePersistentTensor(context, *conv_fwd_pd, &filter_tensor_ptr, in CacheFilter()
1598 void AllocatePersistentTensor(OpKernelContext* context, in AllocatePersistentTensor() function in tensorflow::MklQuantizedConv2DOp
1634 AllocatePersistentTensor(context, *conv_fwd_pd, &bias_tensor_ptr); in CacheBias()
|