OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:kDim1
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
concat_op_test.cc
38
// For the benchmark, we set up two 2-dimensional tensors, each
kDim1
x 'dim'
46
const int
kDim1
= 100;
49
Tensor in0(dt, TensorShape({
kDim1
, dim2}));
51
Tensor in1(dt, TensorShape({
kDim1
, dim2}));
64
((
kDim1
* dim2) + (
kDim1
* dim2)) * sizeof(T));
97
const int
kDim1
= 40000;
104
Tensor in(dt, TensorShape({
kDim1
, dim2}));
116
testing::BytesProcessed(static_cast<int64>(iters) *
kDim1
* dim2 *
132
const int
kDim1
= 100
[
all
...]
segment_reduction_ops_test.cc
137
const int
kDim1
= segments_flat(kNumIndices - 1) + 1;
139
Tensor input(DT_FLOAT, TensorShape({
kDim1
, kDim2}));
152
testing::BytesProcessed(static_cast<int64>(iters) * (
kDim1
* kDim2) *
quantized_concat_op_test.cc
246
// For the benchmark, we set up two 2-dimensional tensors, each
kDim1
x 'dim'
257
const int
kDim1
= 100;
258
TensorShape shape({
kDim1
, dim2});
282
((
kDim1
* dim2) + (
kDim1
* dim2)) * sizeof(T));
Completed in 81 milliseconds