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

  /external/tensorflow/tensorflow/core/kernels/
maxpooling_op_gpu.h 43 const bool include_batch_in_index);
60 const Eigen::GpuDevice& d, const bool include_batch_in_index);
78 const Eigen::GpuDevice& d, const bool include_batch_in_index);
maxpooling_op_gpu.cu.cc 57 // include_batch_in_index: whether to include batch dimension in flattened
71 dtype* top_data, int64* mask, const bool include_batch_in_index) {
91 maxidx = include_batch_in_index ? idx + offset : idx;
145 dtype* top_data, int64* mask, const bool include_batch_in_index) {
166 maxidx = include_batch_in_index ? idx + offset : idx;
234 // include_batch_in_index: whether to include batch dimension in flattened
243 const bool include_batch_in_index) {
246 include_batch_in_index ? 0 : (index / top_offset) * bottom_offset;
364 // include_batch_in_index: whether to include batch dimension in flattened
370 const bool include_batch_in_index) {
    [all...]
maxpooling_op.cc 61 const PoolParameters& params, const bool include_batch_in_index) {
64 context, include_batch_in_index,
66 "SpatialMaxPoolWithArgMaxHelper requires include_batch_in_index "
102 include_batch_in_index](int64 start, int64 limit) {
155 if (include_batch_in_index) {
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
pooling_ops_test.py 845 "Config", ["use_gpu", "include_batch_in_index", "argmax"])
862 include_batch_in_index=config.include_batch_in_index)
878 "Config", ["use_gpu", "include_batch_in_index", "argmax"])
    [all...]
  /external/tensorflow/tensorflow/python/ops/
nn_ops.py     [all...]
nn_grad.py 709 include_batch_in_index=op.get_attr("include_batch_in_index"))
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 192 milliseconds