OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:tensor_idx
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
sparse_conditional_accumulator.h
87
const Tensor*
tensor_idx
= std::get<0>(*tensor);
local
112
tensor_idx
->dims() > 0) {
113
for (int64 i = 0; i <
tensor_idx
->dim_size(0); i++) {
114
if (
tensor_idx
->vec<int64>()(i) >= shape_.dim_size(0)) {
117
"; index is ",
tensor_idx
->vec<int64>()(i), " exceeded ",
depthwise_conv_op_gpu.cu.cc
227
const int
tensor_idx
= thread_pix * in_depth + thread_depth;
251
const int filter_offset =
tensor_idx
+ start_channel;
506
const int
tensor_idx
= thread_depth * in_pixels + thread_pix;
532
const int inout_offset = channel * in_pixels +
tensor_idx
;
[
all
...]
Completed in 62 milliseconds