Home
last modified time | relevance | path

Searched refs:k_ (Results 1 – 25 of 77) sorted by relevance

1234

/external/eigen/bench/tensors/
Dtensor_contract_sycl_bench.cc38 void finalizeBenchmark(Start start, End end, TensorIndex m_, TensorIndex k_, TensorIndex n_ , Tenso… in finalizeBenchmark() argument
41 …std::cout <<"Kernel Name : " << name << ", M : " << m_ << ", N : " << n_ << ", K : " << k_ << " G… in finalizeBenchmark()
42 …static_cast<float>((static_cast<int64_t>(2) * m_ * n_ * k_ * num_iters)/ elapsed_seconds.count()) … in finalizeBenchmark()
43 …out <<"Kernel Name : " << name << ", M : " << m_ << ", N : " << n_ << ", K : " << k_ << " GFLOP/s… in finalizeBenchmark()
44 …static_cast<float>((static_cast<int64_t>(2) * m_ * n_ * k_ * num_iters)/ elapsed_seconds.count()) … in finalizeBenchmark()
49 void contraction(const Device& device_, TensorIndex num_iters, TensorIndex m_, TensorIndex k_, Tens… in contraction() argument
53 a_ = (T *) device_.allocate(m_ * k_ * sizeof(T)); in contraction()
54 b_ = (T *) device_.allocate(k_ * n_ * sizeof(T)); in contraction()
59 device_.memset(a_, 12, m_ * k_ * sizeof(T)); in contraction()
60 device_.memset(b_, 23, k_ * n_ * sizeof(T)); in contraction()
[all …]
Dtensor_benchmarks.h21 : m_(m), k_(k), n_(n), device_(device) { in BenchmarkSuite()
26 : m_(m), k_(m), n_(m), device_(device) { in BenchmarkSuite()
31 : m_(1), k_(k), n_(m), device_(device) { in BenchmarkSuite()
42 eigen_assert(m_ == k_ && k_ == n_); in memcpy()
61 sizes[1] = k_; in typeCasting()
64 sizes[1] = k_ * sizeof(T) / sizeof(int); in typeCasting()
78 finalizeBenchmark(static_cast<int64_t>(m_) * k_ * num_iters); in typeCasting()
82 eigen_assert(m_ == k_ && k_ == n_); in random()
101 eigen_assert(m_ == k_ && k_ == n_); in slicing()
144 input_size[0] = k_; in rowChip()
[all …]
/external/leveldb/util/
Dbloom.cc21 k_ = static_cast<size_t>(bits_per_key * 0.69); // 0.69 =~ ln(2) in BloomFilterPolicy()
22 if (k_ < 1) k_ = 1; in BloomFilterPolicy()
23 if (k_ > 30) k_ = 30; in BloomFilterPolicy()
41 dst->push_back(static_cast<char>(k_)); // Remember # of probes in filter in CreateFilter()
48 for (size_t j = 0; j < k_; j++) { in CreateFilter()
84 size_t k_; member in leveldb::__anon6341b33d0111::BloomFilterPolicy
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtopk_ops.cc99 OP_REQUIRES_OK(ctx, ctx->GetAttr("k", &k_)); in KthOrderStatistic()
100 OP_REQUIRES(ctx, k_ >= 0, errors::InvalidArgument("Need k >= 0, got ", k_)); in KthOrderStatistic()
111 xla::XlaOp k = xla::ConstantR0<int32>(builder, k_); in Compile()
118 int32 k_; member in tensorflow::__anonf9e13b600111::KthOrderStatistic
207 OP_REQUIRES_OK(ctx, ctx->GetAttr("k", &k_)); in TopKUnique()
208 OP_REQUIRES(ctx, k_ >= 0, errors::InvalidArgument("Need k >= 0, got ", k_)); in TopKUnique()
219 auto topk = CreateTopKUnique(builder, input, input_shape, k_, false); in Compile()
225 int k_; member in tensorflow::__anonf9e13b600111::TopKUnique
336 OP_REQUIRES_OK(ctx, ctx->GetAttr("k", &k_)); in TopKWithUnique()
337 OP_REQUIRES(ctx, k_ >= 0, errors::InvalidArgument("Need k >= 0, got ", k_)); in TopKWithUnique()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.h71 k_(k), in MatmulParameters()
95 m_, ", ", n_, ", ", k_, in ToString()
105 return std::make_tuple(transa_, transb_, m_, n_, k_, dtype_, device_id_); in get_data_as_tuple()
112 uint64 k_; variable
Din_topk_op.cc36 OP_REQUIRES_OK(context, context->GetAttr("k", &k_)); in InTopK()
44 int64_t k_value = k_; in Compute()
84 int k_; member in tensorflow::InTopK
Dtopk_op.cc45 OP_REQUIRES_OK(context, context->GetAttr("k", &k_)); in TopK()
47 k_ = -1; in TopK()
52 int k = k_; in Compute()
106 int k_; member in tensorflow::TopK
/external/webrtc/third_party/abseil-cpp/absl/random/
Dzipf_distribution.h68 result_type k() const { return k_; } in k()
73 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
88 IntType k_; variable
152 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
232 assert(ki <= p.k_); in operator()
/external/openscreen/third_party/abseil/src/absl/random/
Dzipf_distribution.h67 result_type k() const { return k_; } in k()
72 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
87 IntType k_; variable
151 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
231 assert(ki <= p.k_); in operator()
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/random/
Dzipf_distribution.h67 result_type k() const { return k_; } in k()
72 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
87 IntType k_; variable
151 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
231 assert(ki <= p.k_); in operator()
Dzipf_distribution_test.cc112 ZipfModel(size_t k, double q, double v) : k_(k), q_(q), v_(v) {} in ZipfModel()
170 hnq_.reserve(std::min(k_, size_t{1000})); in Init()
175 for (size_t i = 0; i < k_; i++) { in Init()
201 const size_t k_; member in __anon761680b20111::ZipfModel
/external/cronet/third_party/abseil-cpp/absl/random/
Dzipf_distribution.h68 result_type k() const { return k_; } in k()
73 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
88 IntType k_; variable
152 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
232 assert(ki <= p.k_); in operator()
Dzipf_distribution_test.cc112 ZipfModel(size_t k, double q, double v) : k_(k), q_(q), v_(v) {} in ZipfModel()
170 hnq_.reserve(std::min(k_, size_t{1000})); in Init()
175 for (size_t i = 0; i < k_; i++) { in Init()
201 const size_t k_; member in __anonab95d1440111::ZipfModel
/external/angle/third_party/abseil-cpp/absl/random/
Dzipf_distribution.h68 result_type k() const { return k_; } in k()
73 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
88 IntType k_; variable
152 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
232 assert(ki <= p.k_); in operator()
Dzipf_distribution_test.cc111 ZipfModel(size_t k, double q, double v) : k_(k), q_(q), v_(v) {} in ZipfModel()
169 hnq_.reserve(std::min(k_, size_t{1000})); in Init()
174 for (size_t i = 0; i < k_; i++) { in Init()
200 const size_t k_; member in __anonfe4f8fe00111::ZipfModel
/external/abseil-cpp/absl/random/
Dzipf_distribution.h68 result_type k() const { return k_; } in k()
73 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
88 IntType k_; variable
152 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
232 assert(ki <= p.k_); in operator()
Dzipf_distribution_test.cc112 ZipfModel(size_t k, double q, double v) : k_(k), q_(q), v_(v) {} in ZipfModel()
170 hnq_.reserve(std::min(k_, size_t{1000})); in Init()
175 for (size_t i = 0; i < k_; i++) { in Init()
201 const size_t k_; member in __anond6b4b4b10111::ZipfModel
/external/libtextclassifier/abseil-cpp/absl/random/
Dzipf_distribution.h67 result_type k() const { return k_; } in k()
72 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
87 IntType k_; variable
151 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
231 assert(ki <= p.k_); in operator()
Dzipf_distribution_test.cc112 ZipfModel(size_t k, double q, double v) : k_(k), q_(q), v_(v) {} in ZipfModel()
170 hnq_.reserve(std::min(k_, size_t{1000})); in Init()
175 for (size_t i = 0; i < k_; i++) { in Init()
201 const size_t k_; member in __anon595d78010111::ZipfModel
/external/tensorflow/third_party/absl/abseil-cpp/absl/random/
Dzipf_distribution.h68 result_type k() const { return k_; } in k()
73 return a.k_ == b.k_ && a.q_ == b.q_ && a.v_ == b.v_;
88 IntType k_; variable
152 : k_(k), q_(q), v_(v), one_minus_q_(1 - q) { in param_type()
232 assert(ki <= p.k_); in operator()
/external/XNNPACK/test/
Dpack-microkernel-tester.h47 this->k_ = k; in k()
52 return this->k_; in k()
120 size_t k_{1};
Dgemm-microkernel-tester.h76 this->k_ = k; in k()
81 return this->k_; in k()
284 size_t k_{1};
/external/webrtc/common_audio/resampler/
Dsinusoidal_linear_chirp_source.cc31 k_ = (max_frequency_ - kMinFrequency) / duration; in SinusoidalLinearChirpSource()
46 destination[i] = sin(2 * M_PI * (kMinFrequency * t + (k_ / 2) * t * t)); in Run()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dapprox_topk_op.cc57 OP_REQUIRES_OK(ctx, ctx->GetAttr("k", &k_)); in ApproxTopKOpBase()
89 ctx->builder(), {ctx->Input(0), iota}, {init_val, init_arg}, k_, in Compile()
105 int64_t k_; member in tensorflow::__anondd4c53d00111::ApproxTopKOpBase
/external/lzma/CPP/Common/
DStringToInt.cpp18 if (res > (k_ ## uintType ## _max) / 10) return 0; \
21 if (res > (k_ ## uintType ## _max) - v) return 0; \

1234