Searched refs:bottom_offset (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 242 const int bottom_offset, dtype* bottom_diff, in MaxPoolBackward() argument
246 include_batch_in_index ? 0 : (index / top_offset) * bottom_offset; in MaxPoolBackward()
369 const int bottom_offset, dtype* bottom_diff, in MaxPoolGradBackward() argument
373 include_batch_in_index ? 0 : (index / bottom_offset) * top_offset; in MaxPoolGradBackward()
458 const int64* mask, const int top_offset, const int bottom_offset, in operator ()() argument
467 output_size, top_diff, mask, top_offset, bottom_offset, bottom_diff, in operator ()()
503 const int64* mask, const int top_offset, const int bottom_offset, in operator ()() argument
510 bottom_offset, bottom_diff, in operator ()()
|
| D | maxpooling_op_gpu.h | 59 const int bottom_offset, T* bottom_diff,
77 const int bottom_offset, T* bottom_diff,
|
| D | maxpooling_op.cc | 1347 const int bottom_offset = in launch() local
1352 bottom_offset, grad_out->flat<T>().data(), context->eigen_gpu_device(), in launch()
1372 const int bottom_offset = in launch() local
1377 bottom_offset, grad_out->flat<T>().data(), context->eigen_gpu_device(), in launch()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional_test.py | 362 for bottom_offset in [-1, -2]:
363 np.testing.assert_allclose(np_output[:, bottom_offset, :, :], 0.)
372 for bottom_offset in [-1, -2]:
373 np.testing.assert_allclose(np_output[:, :, bottom_offset, :], 0.)
|