Home
last modified time | relevance | path

Searched defs:new_values (Results 1 – 25 of 40) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseCsrTensorMath.cu411 __global__ void reduce_sparse_csr_dim0_cuda_kernel(acc_t* new_values, in reduce_sparse_csr_dim0_cuda_kernel()
493 Tensor new_values = std::get<0>(acc_buffer); in reduce_sparse_csr_dim0_cuda_template() local
540 __global__ void reduce_sparse_csr_dim1_cuda_kernel(acc_t* new_values, in reduce_sparse_csr_dim1_cuda_kernel()
584 Tensor new_values = std::get<0>(acc_buffer); in reduce_sparse_csr_dim1_cuda_template() local
635 Tensor new_values, new_values_acc; in reduce_sparse_csr_dim01_cuda_template() local
H A DSoftMax.cu404 auto new_values = in cuda_sparse_coo_softmax() local
408 auto new_values = at::cuda::_softmax(values, dim - sparse_dim + 1, false); in cuda_sparse_coo_softmax() local
/aosp_15_r20/frameworks/base/media/mca/filterfw/native/core/
H A Dvalue.cpp78 int SetPtrValue(Value* value, const BASE* new_values, int count) { in SetPtrValue()
218 int SetIntArrayValue(Value* value, const int* new_values, int count) { in SetIntArrayValue()
222 int SetFloatArrayValue(Value* value, const float* new_values, int count) { in SetFloatArrayValue()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DSparseCsrTensorUtils.h417 Tensor new_values, new_values_acc; variable
435 inline void copy_from_acc_buffer(Tensor& new_values, Tensor& new_values_acc) { in copy_from_acc_buffer()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/
H A Dsubgraph_utils.cpp232 std::unordered_set<Value*>& new_values, in collectNestedUses()
278 std::unordered_set<Value*> new_values; in closedOverValues() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSoftMax.cpp309 auto new_values = in cpu_sparse_coo_softmax() local
313 auto new_values = at::cpu::_softmax(values, dim - sparse_dim + 1, false); in cpu_sparse_coo_softmax() local
H A DSparseCsrTensorMath.cpp1079 Tensor new_values = std::get<0>(acc_buffer); in reduce_sparse_csr_dim0_cpu_template() local
1167 Tensor new_values = std::get<0>(acc_buffer); in reduce_sparse_csr_dim1_cpu_template() local
1262 Tensor new_values; in reduce_sparse_csr_dim01_cpu_template() local
/aosp_15_r20/external/webrtc/modules/audio_processing/agc2/rnn_vad/
H A Dring_buffer.h39 void Push(rtc::ArrayView<const T, S> new_values) { in Push()
H A Dsequence_buffer.h64 void Push(rtc::ArrayView<const T, N> new_values) { in Push()
/aosp_15_r20/external/cldr/tools/cldr-code/src/main/java/org/unicode/cldr/tool/
H A DCldrConfigOption.java19 Set<String> new_values, in show()
H A DSearchXml.java547 Set<String> new_values, in show()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_tensor_variant.h52 void set_values(const Tensor& new_values) { values_ = new_values; } in set_values()
/aosp_15_r20/frameworks/base/tools/aapt2/process/
H A DProductFilter.cpp93 std::vector<std::unique_ptr<ResourceConfigValue>> new_values; in Consume() local
/aosp_15_r20/external/cronet/base/test/
H A Dtest_future.h396 auto new_values = std::make_tuple(std::forward<Types>(values)...); in SetValue() local
/aosp_15_r20/external/protobuf/java/kotlin/src/test/proto/com/google/protobuf/
H A Devil_names_proto3.proto70 repeated string new_values = 26; field
/aosp_15_r20/external/cronet/third_party/protobuf/java/kotlin/src/test/proto/com/google/protobuf/
H A Devil_names_proto3.proto70 repeated string new_values = 26; field
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorConversions.cpp275 const auto new_values = at::native::to( in _to_copy() local
303 const auto new_values = at::native::to( in _to_copy() local
1421 auto new_values = new_values_2d.unflatten(0, rebatch_sizes_nonempty); in sparse_compressed_to_flipped() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc552 SmallVector<ElementValueT, 16> new_values; in ConstFoldBinaryOpDenseDense() local
646 SmallVector<APFloat, 16> new_values; in ConstFoldUnaryOp() local
3069 SmallVector<ValueType, 16> new_values; in BuildConstRangeTensor() local
3274 std::vector<Attribute> *new_values) { in ComputePermutation()
3326 std::vector<Attribute> new_values; in fold() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_tensor.py1382 def with_values(self, new_values): argument
1411 def with_flat_values(self, new_values): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
H A Dtfrt_fallback_async.cc269 SmallVector<Value, 1> new_values; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsparse_tensor.py177 def with_values(self, new_values): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference.cc1209 std::vector<ShapeAndType> new_values(shapes_and_types.size()); in MergeHandleShapesAndTypes() local
1267 std::vector<ShapeAndType> new_values(shapes_and_types.size()); in RelaxHandleShapesAndMergeTypes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
H A Dtfrt_fallback_async.cc363 SmallVector<Value, 1> new_values; in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dpost_quantize.cc259 std::vector<Attribute> new_values; in matchAndRewrite() local
H A Dprepare_quantize_dynamic_range.cc365 std::vector<Eigen::half> new_values; in convertToFloat16Constant() local

12