Searched refs:SetOp (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 92 lru_cache.SetOp(std::to_string(k), new int(k)); in TEST()
|
| D | mkl_util.h | 1721 void SetOp(const string& key, T* op) { in SetOp() function
1801 void SetOp(const string& key, MklPrimitive* op) { in SetOp() function
1803 lru_cache.SetOp(key, op); in SetOp()
1980 this->SetOp(key, op); in SetReorder()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_softmax_op.cc | 187 this->SetOp(key, op); in SetSoftmaxFwd()
|
| D | mkl_pooling_ops_common.h | 207 this->SetOp(key, op); in SetPoolingFwd()
351 this->SetOp(key, op); in SetPoolingBwd()
|
| D | mkl_matmul_ops_common.h | 377 this->SetOp(key, op); in SetMklDnnMatMulFwd()
690 this->SetOp(key, op); in SetMklMatMul()
|
| D | mkl_quantize_op.cc | 207 this->SetOp(key, op); in SetReorder()
|
| D | mkl_slice_op.cc | 307 this->SetOp(key, op); in SetReorder()
|
| D | mkl_conv_grad_input_ops.cc | 299 this->SetOp(key, op); in SetConvBwdInput()
|
| D | mkl_relu_op.cc | 212 this->SetOp(key, op); in SetEltwiseFwd()
424 this->SetOp(key, op); in SetEltwiseBwd()
|
| D | mkl_conv_grad_filter_ops.cc | 355 this->SetOp(key, op); in SetConvBwdFilter()
|
| D | mkl_fused_batch_norm_op.cc | 373 this->SetOp(key, op); in SetBatchNormFwd()
630 this->SetOp(key, op); in SetBatchNormBwd()
|
| D | mkl_concat_op.cc | 435 this->SetOp(key, op); in SetConcatFwd()
|
| D | mkl_conv_ops.cc | 408 this->SetOp(key, op); in SetConvFwd()
|