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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 213 void AllocateOutputTensor(
280 void AllocateOutputTensor(
mkl_conv_grad_filter_ops.cc 538 AllocateOutputTensor(context, bwd_pd, bwd_output_dims, bwd_output_format,
566 void AllocateOutputTensor(
mkl_conv_grad_input_ops.cc 443 AllocateOutputTensor(context, bwd_pd, bwd_output_dims, bwd_output_format,
453 void AllocateOutputTensor(
mkl_conv_ops.cc 614 AllocateOutputTensor(context, conv_prim_desc, output_dims_mkl_order,
635 AllocateOutputTensor(context, conv_prim_desc, output_dims_mkl_order,
665 void AllocateOutputTensor(
    [all...]
mkl_lrn_op.cc 821 AllocateOutputTensor(context, lrn_prim_desc, input_dims, input_format,
    [all...]
mkl_avgpooling_op.cc 520 this->AllocateOutputTensor(context, pool_prim_desc, output_dims_mkl_order,
626 this->AllocateOutputTensor(
mkl_maxpooling_op.cc 564 this->AllocateOutputTensor(context, pool_fwd_desc, output_dims_mkl_order,
690 this->AllocateOutputTensor(context, pool_bkwd_prim_desc,
    [all...]

Completed in 412 milliseconds