| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_filter_ops.cc | 327 dnnLayout_t mkl_lt_internal_input, mkl_lt_internal_out_backprop; local
332 mkl_lt_internal_input = nullptr;
342 &mkl_lt_internal_input, prim_conv_bwdfilter, dnnResourceSrc),
345 !dnnLayoutCompare_F32(mkl_lt_internal_input, lt_input);
348 mkl_lt_internal_input),
350 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input,
357 dnnLayoutDelete_F32(mkl_lt_internal_input);
|
| mkl_conv_ops.cc | 381 mkl_lt_internal_input; local
389 mkl_lt_internal_input = nullptr;
398 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&mkl_lt_internal_input,
402 !dnnLayoutCompare_F32(mkl_lt_internal_input, lt_input);
405 mkl_lt_internal_input),
407 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input,
414 dnnLayoutDelete_F32(mkl_lt_internal_input);
[all...] |
| mkl_fused_batch_norm_op.cc | 282 dnnLayout_t mkl_lt_internal_input = nullptr; local
289 &mkl_lt_internal_input, mkl_prim_batchnorm, dnnResourceSrc),
292 !dnnLayoutCompare_F32(mkl_lt_internal_input, mkl_lt_input);
295 mkl_lt_internal_input),
297 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input,
304 dnnLayoutDelete_F32(mkl_lt_internal_input);
577 dnnLayout_t mkl_lt_internal_input = nullptr; local
585 &mkl_lt_internal_input, mkl_prim_batchnorm_bwd, dnnResourceSrc),
588 !dnnLayoutCompare_F32(mkl_lt_internal_input, mkl_lt_input);
591 mkl_lt_internal_input),
[all...] |
| mkl_concat_op.cc | 447 dnnLayout_t mkl_lt_internal_input; local
451 &mkl_lt_internal_input, prim_concat,
455 if (!dnnLayoutCompare_F32(lt_inputs[i], mkl_lt_internal_input)) {
457 lt_inputs[i], mkl_lt_internal_input),
460 AllocTmpBuffer(context, &mkl_tmp_tensors[i], mkl_lt_internal_input,
477 CHECK_EQ(dnnLayoutDelete_F32(mkl_lt_internal_input), E_SUCCESS);
[all...] |
