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

  /external/tensorflow/tensorflow/core/kernels/
bias_op_gpu.cu.cc 129 AccT* s_data = reinterpret_cast<AccT*>(s_buf); local
131 s_data[index] = AccT(0);
138 CudaAtomicAdd(s_data + bias_offset, AccT(ldg(output_backprop + index)));
143 CudaAtomicAdd(bias_backprop + index, T(s_data[index]));
155 __shared__ AccT s_data[kSDataSize]; local
157 s_data[index] = AccT(0);
179 CudaAtomicAdd(s_data + bias_offset, sum);
186 AccT data = s_data[thread_index];
  /external/tensorflow/tensorflow/core/lib/gtl/
edit_distance.h 54 const T* s_data = s.data(); local
79 const int replacement_cost = cmp(s_data[i - 1], t_data[j - 1]) ? 0 : 1;
  /external/tensorflow/tensorflow/core/ops/
math_ops.cc 1103 ShapeHandle s_data = c->input(0); local
    [all...]

Completed in 5706 milliseconds