OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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