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

  /external/tensorflow/tensorflow/core/kernels/
spacetobatch_functor.cc 42 const int64* batch_tensor_shape,
44 for (int64 batch_tensor_pos = 0; batch_tensor_pos < batch_tensor_shape[0];
52 pad_start + 1, block_offsets + 1, batch_tensor_shape + 1,
73 const int64* batch_tensor_shape,
106 batch_tensor_shape[NUM_BLOCK_DIMS]; local
111 batch_tensor_shape[block_dim] = batch_tensor.dimension(block_dim + 1);
146 block_offsets, batch_tensor_shape, &batch_tensor_strides[1],
spacetobatch_functor_gpu.cu.cc 36 int32 batch_tensor_shape[NUM_BLOCK_DIMS + 2]; member in struct:tensorflow::S2BParameters
57 remaining_batch_tensor_idx % args.batch_tensor_shape[dim];
58 remaining_batch_tensor_idx /= args.batch_tensor_shape[dim];
64 int32 space_tensor_stride = args.batch_tensor_shape[NUM_BLOCK_DIMS + 1];
135 args.batch_tensor_shape[dim] = batch_tensor.dimension(dim);
136 total_count *= args.batch_tensor_shape[dim];

Completed in 573 milliseconds