| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_filter_ops.cc | 221 Tensor mkl_tmp_input_buf_tensor, mkl_tmp_out_backprop_buf_tensor; variable
223 mkl_context.MklPrepareInputs(context, &mkl_tmp_input_buf_tensor,
323 Tensor* mkl_tmp_input_buf_tensor,
350 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input,
|
| mkl_lrn_op.cc | 146 Tensor mkl_tmp_input_buf_tensor; variable
147 mkl_context.MklPrepareLRNInputs(context, &mkl_tmp_input_buf_tensor);
184 Tensor* mkl_tmp_input_buf_tensor) {
201 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, lt_internal_input,
473 Tensor mkl_tmp_input_buf_tensor, mkl_tmp_image_buf_tensor, variable
476 mkl_context.MklPrepareLRNGradInput(context, &mkl_tmp_input_buf_tensor,
533 Tensor* mkl_tmp_input_buf_tensor,
565 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, lt_bdw_input,
578 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, lt_bdw_input,
[all...] |
| mkl_conv_ops.cc | 318 Tensor mkl_tmp_input_buf_tensor, mkl_tmp_filter_buf_tensor, variable
320 mkl_context.MklPrepareConvolutionInputs(context, &mkl_tmp_input_buf_tensor,
374 Tensor* mkl_tmp_input_buf_tensor,
407 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input,
[all...] |
| mkl_relu_op.cc | 203 Tensor* mkl_tmp_input_buf_tensor) {
214 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, lt_grad,
330 Tensor mkl_tmp_input_buf_tensor; local
331 mkl_context.MklPrepareReluGradInputs(context, &mkl_tmp_input_buf_tensor);
[all...] |
| mkl_fused_batch_norm_op.cc | 123 Tensor mkl_tmp_input_buf_tensor, mkl_tmp_scale_shift_buf_tensor; variable
124 mkl_context.MklPrepareContextInputs(context, &mkl_tmp_input_buf_tensor,
278 Tensor* mkl_tmp_input_buf_tensor,
297 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input,
408 Tensor mkl_tmp_input_buf_tensor, mkl_tmp_outbackprop_buf_tensor, variable
410 mkl_context.MklPrepareContextInputs(context, &mkl_tmp_input_buf_tensor,
572 Tensor* mkl_tmp_input_buf_tensor,
593 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input,
[all...] |
