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

  /external/tensorflow/tensorflow/core/kernels/
spacetobatch_functor.cc 43 const int64* batch_tensor_strides, T* batch_tensor_ptr) {
53 batch_tensor_strides + 1, batch_tensor_ptr);
58 batch_tensor_ptr[i] = static_cast<T>(0);
62 batch_tensor_ptr += batch_tensor_strides[0];
74 const int64* batch_tensor_strides, T* batch_tensor_ptr) {
77 batch_tensor_ptr[i] = space_tensor_ptr[i];
79 space_tensor_ptr[i] = batch_tensor_ptr[i];
128 T* batch_tensor_ptr = const_cast<T*>(batch_tensor.data()); local
147 batch_tensor_ptr + batch_tensor_b * batch_tensor_strides[0]);
spacetobatch_functor_gpu.cu.cc 49 S2BParameters<NUM_BLOCK_DIMS> args, T* batch_tensor_ptr) {
79 batch_tensor_ptr[batch_tensor_idx] = static_cast<T>(0);
88 batch_tensor_ptr[batch_tensor_idx] =
92 ldg(batch_tensor_ptr + batch_tensor_idx);

Completed in 56 milliseconds