HomeSort by relevance Sort by last modified time
    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