Home
last modified time | relevance | path

Searched defs:weights_ (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMultiMarginLoss.cu165 const std::optional<Tensor> &weights_, int64_t reduction, Tensor& out_) { in multi_margin_loss_cuda_out()
299 const Scalar &p_, const Scalar &margin_, const std::optional<Tensor> &weights_, in multi_margin_loss_cuda_backward_out()
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/src/core/ext/filters/client_channel/lb_policy/weighted_round_robin/
Dstatic_stride_scheduler.h66 std::vector<uint16_t> weights_; variable
/aosp_15_r20/external/grpc-grpc/src/core/load_balancing/weighted_round_robin/
H A Dstatic_stride_scheduler.h66 std::vector<uint16_t> weights_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
H A Dfully_connected_op_builder.h50 TfLiteTensor* weights_ = nullptr; variable
H A Dconvolution_op_builder.h77 TfLiteTensor* weights_ = nullptr; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmfcc_mel_filterbank.h52 std::vector<double> weights_; // Right-side weight for this fft bin. variable
H A Drange_sampler.h232 std::vector<float> weights_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dmfcc_mel_filterbank.h52 std::vector<double> weights_; // Right-side weight for this fft bin. variable
/aosp_15_r20/external/webrtc/modules/audio_processing/agc2/rnn_vad/
H A Drnn_gru.h60 const std::vector<float> weights_; variable
H A Drnn_fc.h62 const std::vector<float> weights_; variable
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/
DRNN.h74 const RunTimeOperandInfo* weights_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected_test.cc250 int weights_; member in tflite::__anonaa112c3c0111::BaseFullyConnectedOpModel
457 int weights_; member in tflite::__anonaa112c3c0111::HybridFullyConnectedOpModel
1602 int weights_; member in tflite::__anonaa112c3c0111::SparseFullyConnectedOpModel
H A Dembedding_lookup_sparse_test.cc85 int weights_; member in tflite::__anon878db3e30111::EmbeddingLookupSparseOpModel
H A Dquant_basic_lstm_test.cc146 int weights_; member in tflite::__anon4e71b0670111::QuantizedLSTMOpModel
H A Dbasic_rnn_test.cc221 int weights_; member in tflite::__anone1009ead0111::RNNOpModel
H A Dunidirectional_sequence_rnn_test.cc236 int weights_; member in tflite::__anon6f5dbcbd0111::UnidirectionalRNNOpModel
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dweights.h280 TRT_ShapedWeights weights_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dmatmul_test.cc97 int weights_; member in tflite::FullyConnectedOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dquant_lstm_sup_test.cc152 const std::vector<uint8_t> weights_; member in __anon05eb701c0111::QuantLstmWeightDecompTest
H A Dnnapi_delegate_signed_quantization_test.cc123 int weights_; member in tflite::__anonf5e0ceac0111::HybridFullyConnectedOpModel
H A Dnnapi_delegate_test.cc1320 int weights_; member in tflite::__anon1a74836d0111::FullyConnectedOpModel
2860 int weights_; member in tflite::__anon1a74836d0111::RNNOpModel
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
H A Dquantization_driver.cc386 llvm::DenseSet<Operation *> weights_; member in mlir::quant::__anon95f8f9910111::QuantizationDriver
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1808 ParamsRegions weights_; member in stream_executor::gpu::MIOpenRnnParamsDescriptor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1225 ParamsRegions weights_; member in stream_executor::gpu::__anon96353b0e0311::CudnnRnnParamsDescriptor