/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | MultiMarginLoss.cu | 165 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/ |
D | static_stride_scheduler.h | 66 std::vector<uint16_t> weights_; variable
|
/aosp_15_r20/external/grpc-grpc/src/core/load_balancing/weighted_round_robin/ |
H A D | static_stride_scheduler.h | 66 std::vector<uint16_t> weights_; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
H A D | fully_connected_op_builder.h | 50 TfLiteTensor* weights_ = nullptr; variable
|
H A D | convolution_op_builder.h | 77 TfLiteTensor* weights_ = nullptr; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | mfcc_mel_filterbank.h | 52 std::vector<double> weights_; // Right-side weight for this fft bin. variable
|
H A D | range_sampler.h | 232 std::vector<float> weights_; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
H A D | mfcc_mel_filterbank.h | 52 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 D | rnn_gru.h | 60 const std::vector<float> weights_; variable
|
H A D | rnn_fc.h | 62 const std::vector<float> weights_; variable
|
/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
D | RNN.h | 74 const RunTimeOperandInfo* weights_; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | fully_connected_test.cc | 250 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 D | embedding_lookup_sparse_test.cc | 85 int weights_; member in tflite::__anon878db3e30111::EmbeddingLookupSparseOpModel
|
H A D | quant_basic_lstm_test.cc | 146 int weights_; member in tflite::__anon4e71b0670111::QuantizedLSTMOpModel
|
H A D | basic_rnn_test.cc | 221 int weights_; member in tflite::__anone1009ead0111::RNNOpModel
|
H A D | unidirectional_sequence_rnn_test.cc | 236 int weights_; member in tflite::__anon6f5dbcbd0111::UnidirectionalRNNOpModel
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
H A D | weights.h | 280 TRT_ShapedWeights weights_; variable
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
H A D | matmul_test.cc | 97 int weights_; member in tflite::FullyConnectedOpModel
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
H A D | quant_lstm_sup_test.cc | 152 const std::vector<uint8_t> weights_; member in __anon05eb701c0111::QuantLstmWeightDecompTest
|
H A D | nnapi_delegate_signed_quantization_test.cc | 123 int weights_; member in tflite::__anonf5e0ceac0111::HybridFullyConnectedOpModel
|
H A D | nnapi_delegate_test.cc | 1320 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 D | quantization_driver.cc | 386 llvm::DenseSet<Operation *> weights_; member in mlir::quant::__anon95f8f9910111::QuantizationDriver
|
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
H A D | rocm_dnn.cc | 1808 ParamsRegions weights_; member in stream_executor::gpu::MIOpenRnnParamsDescriptor
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
H A D | cuda_dnn.cc | 1225 ParamsRegions weights_; member in stream_executor::gpu::__anon96353b0e0311::CudnnRnnParamsDescriptor
|