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

  /external/tensorflow/tensorflow/core/kernels/
mkl_avgpooling_op.cc 125 MklShape mkl_out_shape; variable
126 mkl_out_shape.SetMklTensor(true);
127 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst);
128 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim,
131 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_);
134 mkl_out_shape.GetMklLayout())) /
138 mkl_out_shape);
315 MklShape mkl_out_shape; variable
316 mkl_out_shape.SetMklTensor(true);
317 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_bwd
    [all...]
mkl_maxpooling_op.cc 106 MklShape mkl_out_shape, mkl_workspace_shape; variable
107 mkl_out_shape.SetMklTensor(true);
108 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst);
109 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim,
112 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_);
116 mkl_out_shape.GetMklLayout())) /
119 mkl_out_shape);
    [all...]

Completed in 90 milliseconds