Searched refs:SetOp (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 95 lru_cache.SetOp(std::to_string(k), new int(k)); in TEST()
|
| D | mkl_util.h | 2100 void SetOp(const string& key, T* op) { in SetOp() function
2180 void SetOp(const string& key, MklPrimitive* op) { in SetOp() function
2182 lru_cache.SetOp(key, op); in SetOp()
2351 this->SetOp(key, op); in SetReorder()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_pooling_ops_common.h | 206 this->SetOp(key, op); in SetPoolingFwd()
357 this->SetOp(key, op); in SetPoolingBwd()
|
| D | mkl_slice_op.cc | 300 this->SetOp(key, op); in SetReorder()
|
| D | mkl_fused_batch_norm_op.cc | 267 this->SetOp(key, op); in SetBatchNormFwd()
491 this->SetOp(key, op); in SetBatchNormBwd()
|
| D | mkl_relu_op.cc | 219 this->SetOp(key, op); in SetEltwiseFwd()
441 this->SetOp(key, op); in SetEltwiseBwd()
|
| D | mkl_conv_grad_input_ops.cc | 294 this->SetOp(key, op); in SetConvBwdInput()
|
| D | mkl_conv_grad_filter_ops.cc | 356 this->SetOp(key, op); in SetConvBwdFilter()
|
| D | mkl_conv_ops.cc | 406 this->SetOp(key, op); in SetConvFwd()
|