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

  /external/tensorflow/tensorflow/core/kernels/
spacetobatch_functor.cc 46 const int64 space_tensor_pos = local
48 if (space_tensor_pos >= 0 && space_tensor_pos < space_tensor_shape[0]) {
50 space_tensor_ptr + space_tensor_pos * space_tensor_strides[0],
spacetobatch_functor_gpu.cu.cc 72 int32 space_tensor_pos = local
75 if (space_tensor_pos < 0 ||
76 space_tensor_pos >= args.space_tensor_spatial_shape[block_dim]) {
83 space_tensor_idx += space_tensor_stride * space_tensor_pos;

Completed in 158 milliseconds