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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.cc 28 void MklPoolParameters::Init(OpKernelContext* context,
47 void MklPoolParameters::Init(OpKernelContext* context,
62 void MklPoolParameters::Init(OpKernelContext* context,
77 void MklPoolParameters::Init(OpKernelContext* context,
152 const MklPoolParameters& params,
mkl_pooling_ops_common.h 37 struct MklPoolParameters {
63 MklPoolParameters()
144 void GetOutputDims(const MklPoolParameters& mkl_pool_params,
154 MklPoolParameters* pool_params,
197 MklPoolParameters* pool_params,
332 memory::dims* original_input_dims_nchw, MklPoolParameters* pool_params,
352 const MklPoolParameters& pool_params,
417 const MklPoolParameters& params,
mkl_avgpooling_op.cc 79 MklPoolParameters pool_params;
257 MklPoolParameters pool_params;
457 MklPoolParameters pool_params;
574 MklPoolParameters pool_params;
654 MklPoolParameters* pool_params, TensorShape* input_tensor_shape) {
mkl_maxpooling_op.cc 83 MklPoolParameters pool_params;
249 MklPoolParameters pool_params;
525 MklPoolParameters pool_params;
647 MklPoolParameters pool_params;
726 MklPoolParameters* pool_params, TensorShape* input_tensor_shape) {
    [all...]

Completed in 70 milliseconds