Searched refs:GetTensorBuffer (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1330 inline void* GetTensorBuffer(const Tensor* tensor) const { in GetTensorBuffer() function
1356 SetUsrMem(dim, fm, GetTensorBuffer(tensor)); in SetUsrMem()
1391 SetUsrMem(dim, strides, GetTensorBuffer(tensor)); in SetUsrMem()
1397 SetUsrMem(md, GetTensorBuffer(tensor)); in SetUsrMem()
1445 SetUsrMemDataHandle(GetTensorBuffer(tensor), t_stream);
1648 return CheckReorderToOpMem(op_md, GetTensorBuffer(reorder_tensor), net, in CheckReorderToOpMem()
1659 return CheckReorderToOpMem(op_pd, GetTensorBuffer(reorder_tensor),
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 528 void* weight_oi_t_data = weight.GetTensorBuffer(&weight_oi_); in CacheWeight()
|
| D | mkl_conv_ops.cc | 874 filter.GetTensorBuffer(filter_out_tensor), cpu_engine_, in Compute()
1316 void* cached_filter_data = filter.GetTensorBuffer(filter_tensor_ptr); in CacheFilter()
|