HomeSort by relevance Sort by last modified time
    Searched refs:SetMklTensor (Results 1 - 18 of 18) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_softmax_op.cc 109 output_mkl_shape.SetMklTensor(true);
116 output_mkl_shape.SetMklTensor(false);
mkl_relu_op.cc 81 mkl_context.output_shape.SetMklTensor(false);
123 mkl_context.output_shape.SetMklTensor(true);
137 mkl_context.output_shape.SetMklTensor(false);
287 mkl_context.output_shape.SetMklTensor(false);
336 mkl_context.output_shape.SetMklTensor(true);
358 mkl_context.output_shape.SetMklTensor(false);
426 dnn_shape_dst.SetMklTensor(true);
435 dnn_shape_dst.SetMklTensor(false);
583 dnn_shape_diff_src.SetMklTensor(true);
592 dnn_shape_diff_src.SetMklTensor(false)
    [all...]
mkl_aggregate_ops.cc 78 mkl_context.output_shape.SetMklTensor(false);
99 mkl_context.output_shape.SetMklTensor(false);
140 mkl_context.output_shape.SetMklTensor(true);
161 mkl_context.output_shape.SetMklTensor(false);
337 mkl_shape_dst.SetMklTensor(false);
351 mkl_shape_dst.SetMklTensor(false);
461 output_mkl_shape.SetMklTensor(true);
476 output_mkl_shape.SetMklTensor(false);
mkl_fused_batch_norm_op.cc 131 mkl_shape_output.SetMklTensor(true);
150 mkl_shape_batch_mean.SetMklTensor(false);
156 mkl_shape_batch_variance.SetMklTensor(false);
177 mkl_shape_saved_mean.SetMklTensor(false);
186 mkl_shape_saved_variance.SetMklTensor(false);
418 mkl_shape_in_backprop.SetMklTensor(true);
    [all...]
mkl_reshape_op.cc 113 mkl_shape_output.SetMklTensor(false);
254 mkl_shape_output.SetMklTensor(false);
286 mkl_shape_output.SetMklTensor(true);
mkl_conv_ops.cc 192 mkl_output_mkl_shape.SetMklTensor(false);
201 mkl_output_mkl_shape.SetMklTensor(false);
276 mkl_output_mkl_shape.SetMklTensor(true);
293 mkl_filter_output_mkl_shape.SetMklTensor(true);
553 output_mkl_shape.SetMklTensor(false);
558 filter_mkl_shape.SetMklTensor(false);
675 output_mkl_shape.SetMklTensor(true);
699 filter_mkl_shape.SetMklTensor(true);
    [all...]
mkl_input_conversion_op.cc 192 mkl_output_mkl_shape.SetMklTensor(true);
317 mkl_output_mkl_shape.SetMklTensor(true);
425 mkl_output_mkl_shape.SetMklTensor(true);
mkl_lrn_op.cc 219 mkl_output_mkl_shape.SetMklTensor(true);
236 mkl_workspace_mkl_shape.SetMklTensor(false);
301 mkl_output_mkl_shape.SetMklTensor(false);
306 mkl_workspace_mkl_shape.SetMklTensor(false);
449 mkl_output_mkl_shape.SetMklTensor(true);
671 mkl_output_mkl_shape.SetMklTensor(false);
    [all...]
mkl_concat_op.cc 388 mkl_output_mkl_shape.SetMklTensor(true);
514 mkl_tensor_mkl_shape.SetMklTensor(false);
545 mkl_tensor_mkl_shape.SetMklTensor(false);
751 dnn_shape_dst.SetMklTensor(true);
759 dnn_shape_dst.SetMklTensor(false);
797 dnn_shape_output.SetMklTensor(false);
    [all...]
mkl_conv_grad_filter_ops.cc 228 mkl_context.grad_filter_shape.SetMklTensor(false);
580 output_mkl_shape.SetMklTensor(false);
598 bias_grad_mkl_shape.SetMklTensor(false);
mkl_pooling_ops_common.h 222 output_mkl_shape.SetMklTensor(true);
289 output_mkl_shape.SetMklTensor(true);
mkl_conv_grad_bias_ops.cc 89 output_mkl_shape.SetMklTensor(false);
mkl_maxpooling_op.cc 106 mkl_out_shape.SetMklTensor(true);
124 mkl_workspace_shape.SetMklTensor(false);
277 mkl_output_shape.SetMklTensor(true);
596 workspace_mkl_shape.SetMklTensor(false);
    [all...]
mkl_conv_grad_input_ops.cc 208 mklOutputShape.SetMklTensor(true);
465 output_mkl_shape.SetMklTensor(true);
mkl_avgpooling_op.cc 126 mkl_out_shape.SetMklTensor(true);
316 mkl_out_shape.SetMklTensor(true);
471 output_mkl_shape.SetMklTensor(false);
mkl_conv_ops.h 390 output_mkl_shape.SetMklTensor(false);
  /external/tensorflow/tensorflow/core/util/
mkl_util_test.cc 34 a_mkldnn_shape.SetMklTensor(true);
46 b_mkldnn_shape.SetMklTensor(true);
mkl_util.h 83 void SetMklTensor(const bool isMklTensor) { isMklTensor_ = isMklTensor; }
419 inline void SetMklTensor(bool is_mkl_tensor) {
    [all...]

Completed in 675 milliseconds