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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 204 ? input_mkl_shape.GetMklLayout()
346 ? original_input_mkl_shape.GetMklLayout()
358 ? original_output_mkl_shape.GetMklLayout()
371 ? input_gradient_mkl_shape.GetMklLayout()
mkl_softmax_op.cc 80 ? src_mkl_shape.GetMklLayout()
mkl_aggregate_ops.cc 154 mkl_context.output_shape.GetMklLayout())) /
390 md1 = src1_mkl_shape.GetMklLayout();
407 md2 = src2_mkl_shape.GetMklLayout();
411 md1 = src1_mkl_shape.GetMklLayout();
412 md2 = src2_mkl_shape.GetMklLayout();
mkl_relu_op.cc 131 mkl_context.output_shape.GetMklLayout())) /
352 mkl_context.output_shape.GetMklLayout())) /
404 src_md = dnn_shape_src.GetMklLayout();
523 src_md = dnn_shape_src.GetMklLayout();
535 diff_dst_md = dnn_shape_diff_dst.GetMklLayout();
548 src_md = dnn_shape_src.GetMklLayout();
549 diff_dst_md = dnn_shape_diff_dst.GetMklLayout();
    [all...]
mkl_input_conversion_op.cc 297 auto input0_md = input_shape_0.GetMklLayout();
298 auto input1_md = input_shape_1.GetMklLayout();
431 auto output_mkl_md = mkl_shape->GetMklLayout();
mkl_tfconv_op.h 93 auto input_mkl_md = input_shape.GetMklLayout();
mkl_reshape_op.cc 153 auto input_mkl_md = mkl_shape_input.GetMklLayout();
244 auto input_mkl_md = mkl_shape_input.GetMklLayout();
mkl_conv_ops.h 430 ? input_mkl_shape.GetMklLayout()
435 ? filter_mkl_shape.GetMklLayout()
467 auto outbprop_md = outbprop_mkl_shape.GetMklLayout();
mkl_concat_op.cc 398 static_cast<dnnLayout_t>(mkl_output_mkl_shape.GetMklLayout())) /
698 is_mkl_tensor ? input_shapes[k].GetMklLayout() :
725 (memory::format)input_shapes[0].GetMklLayout().data.format);
767 dnn_shape_dst.IsMklTensor() ? dnn_shape_dst.GetMklLayout() : dst_md;
    [all...]
mkl_conv_ops.cc 286 static_cast<dnnLayout_t>(mkl_output_mkl_shape.GetMklLayout())) /
306 mkl_filter_output_mkl_shape.GetMklLayout())) /
575 ? src_mkl_shape.GetMklLayout()
581 ? filter_mkl_shape.GetMklLayout()
    [all...]
mkl_fused_batch_norm_op.cc 140 mkl_shape_output.GetMklLayout())) /
428 static_cast<dnnLayout_t>(mkl_shape_in_backprop.GetMklLayout())) /
    [all...]
mkl_avgpooling_op.cc 134 mkl_out_shape.GetMklLayout())) /
325 mkl_out_shape.GetMklLayout())) /
mkl_maxpooling_op.cc 115 mkl_out_shape.GetMklLayout())) /
287 static_cast<dnnLayout_t>(mkl_output_shape.GetMklLayout())) /
    [all...]
mkl_lrn_op.cc 227 static_cast<dnnLayout_t>(mkl_output_mkl_shape.GetMklLayout())) /
245 mkl_workspace_mkl_shape.GetMklLayout())) /
463 static_cast<dnnLayout_t>(mkl_output_mkl_shape.GetMklLayout())) /
    [all...]
mkl_conv_grad_input_ops.cc 218 mklOutputShape.GetMklLayout())) /
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 151 const dnnLayout_t GetMklLayout() const { return mklLayout_; }
401 CompareMklDnnLayouts(this->GetMklLayout(),
402 input_shape.GetMklLayout());
513 inline const memory::desc GetMklLayout() const {
552 return IsMklTensor() ? GetMklLayout() : GetTfLayout();
    [all...]

Completed in 244 milliseconds