HomeSort by relevance Sort by last modified time
    Searched defs:tensor_in_batch (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
pooling_ops_3d.h 55 int tensor_in_batch; member in struct:tensorflow::Pool3dParameters
pooling_ops_common.h 54 int tensor_in_batch; member in struct:tensorflow::PoolParameters
178 params.tensor_in_batch);
181 params.out_width * params.out_height * params.tensor_in_batch);
189 // depth by (tensor_in_cols * tensor_in_rows * tensor_in_batch)
191 // depth by (out_width * out_height * tensor_in_batch)
254 params.tensor_in_batch, shard_cost, shard);
274 params.tensor_in_batch, params.tensor_in_rows, params.tensor_in_cols,
425 params.tensor_in_batch);
428 params.out_width * params.out_height * params.tensor_in_batch);
436 // depth by (tensor_in_cols * tensor_in_rows * tensor_in_batch)
    [all...]
mkl_pooling_ops_common.h 370 int tensor_in_batch; member in struct:tensorflow::MklPoolParameters
401 tensor_in_batch(0),
494 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch,
500 *output_dims_mkl_order = {mkl_pool_params.tensor_in_batch,
565 output_dims_NHWC_order = {pool_params->tensor_in_batch,

Completed in 112 milliseconds