Home
last modified time | relevance | path

Searched refs:input_vec (Results 1 – 22 of 22) sorted by relevance

/external/federated-compute/fcp/secagg/client/
Dsecagg_client_state.cc100 auto input_vec = input_map.find(vector_spec.name()); in ValidateInput() local
101 if (input_vec == input_map.end() || in ValidateInput()
102 input_vec->second.modulus() != vector_spec.modulus() || in ValidateInput()
103 input_vec->second.num_elements() != vector_spec.length()) { in ValidateInput()
Dsecagg_client_r2_masked_input_coll_input_set_state_test.cc416 std::vector<uint64_t> input_vec = {2, 4, 6, 8}; in TEST() local
421 sum_vec.push_back((mask_vec[i] + input_vec[i]) % bit_width_bound); in TEST()
510 std::vector<uint64_t> input_vec = {2, 4, 6, 8}; in TEST() local
515 sum_vec.push_back((mask_vec[i] + input_vec[i]) % bit_width_bound); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dstring_split_op.cc181 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local
182 const int64_t batch_size = input_vec.dimension(0); in Compute()
203 skip_empty_ ? Split(input_vec(i), delimiter, str_util::SkipEmpty()) in Compute()
204 : Split(input_vec(i), delimiter, str_util::AllowEmpty()); in Compute()
256 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local
257 const int64_t batch_size = input_vec.dimension(0); in Compute()
275 std::vector<StringPiece> parts = SplitV2(input_vec(i), sep, maxsplit_); in Compute()
Dsegment_reduction_ops_gpu.cu.h287 const Tvec* __restrict__ input_vec, // [nouter or any, ninner_vec]
314 x_ok && y_ok ? input_vec[input_idx] : Tvec(initial_value);
358 const Tvec* input_vec, // [nouter or any, ninner_vec]
390 is_sqrtn, input_vec, segment_offsets, indices, weights, output_vec);
452 LookupAndScaleAndCastInputsFunctor(const Tvec* input_vec,
455 : input_vec_(input_vec), indices_(indices), weights_(weights) {}
481 MakeLookupAndScaleAndCastInputsIterator(const Tvec* input_vec,
485 typename CastIteratorTy::FunctorTy functor(input_vec, indices, weights);
496 const Tvec* input_vec, // [nouter or any]
521 input_vec, indices, weights);
[all …]
Dunicode_ops.cc409 const auto& input_vec = input_tensor->flat<tstring>(); in Compute() local
430 for (int i = 0; i < input_vec.size(); ++i) { in Compute()
431 const string& input = input_vec(i); in Compute()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc121 OpKernelContext* ctx, const DataVec* input_vec) { in SetTrtEngineInputs() argument
124 int n_inputs = ctx ? ctx->num_inputs() : (input_vec ? input_vec->size() : 0); in SetTrtEngineInputs()
127 const Tensor& input_tensor = ctx ? ctx->input(i) : input_vec->at(i).tensor; in SetTrtEngineInputs()
136 ctx ? StrCat(IONamePrefixes::kInputPHName, i) : input_vec->at(i).name; in SetTrtEngineInputs()
Dtrt_engine_utils.h50 const DataVec* input_vec = nullptr);
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcollective_ops_test.cc342 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
343 absl::c_iota(input_vec, 0); in XLA_TEST_F()
344 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F()
364 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
365 absl::c_iota(input_vec, 0); in XLA_TEST_F()
366 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F()
457 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
458 absl::c_iota(input_vec, 0); in XLA_TEST_F()
459 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F()
472 input_vec_doubled.reserve(input_vec.size()); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show_multi.cc175 std::vector<string> input_vec; in FormatInputShapes() local
178 input_vec.push_back(absl::StrFormat("%d:unknown", s.first)); in FormatInputShapes()
180 input_vec.push_back( in FormatInputShapes()
185 absl::StrFormat("input_type: %s", absl::StrJoin(input_vec, ",\t")); in FormatInputShapes()
/external/libxaac/fuzzer/
Dxaac_enc_fuzzer.cpp557 IA_ERRORCODE ia_enhaacplus_enc_pcm_data_read(std::vector<WORD8> input_vec, UWORD32 num_samples, in ia_enhaacplus_enc_pcm_data_read() argument
566 data[channel_no][sample_no] = *input_vec.data(); in ia_enhaacplus_enc_pcm_data_read()
618 std::vector<WORD8> input_vec = fuzzed_data->ConsumeBytes<WORD8>(input_size); in ixheaace_calculate_loudness_measure() local
620 ia_enhaacplus_enc_pcm_data_read(input_vec, input_size, pstr_in_cfg->i_channels, samples); in ixheaace_calculate_loudness_measure()
724 std::vector<WORD8> input_vec = fuzzed_data.ConsumeBytes<WORD8>(input_size); in LLVMFuzzerTestOneInput() local
725 if (input_vec.size()) { in LLVMFuzzerTestOneInput()
726 memcpy(pb_inp_buf, input_vec.data(), input_vec.size()); in LLVMFuzzerTestOneInput()
/external/pytorch/torch/csrc/distributed/c10d/
DOps.cpp270 auto input_vec = inputs.vec(); \
272 ->allgather_into_tensor_coalesced(output_vec, input_vec); \
334 auto input_vec = inputs.vec(); \
338 input_vec, \
/external/crosvm/devices/src/virtio/fs/
Dconfig.rs409 let input_vec = vec![ in test_process_ugid_map_valid() localVariable
428 let result = process_ugid_map(input_vec).unwrap(); in test_process_ugid_map_valid()
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py246 input_vec = op.inputs[0]
248 input_rank = array_ops.rank(input_vec)
251 return gen_xla_ops.xla_dynamic_update_slice(array_ops.zeros_like(input_vec),
257 array_ops.shape(input_vec) - slice_size - begin_vec, shape)
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5639 const int8x16_t input_vec = vld1q_s8(input_data + i); variable
5640 const int16x8_t first_half = vmovl_s8(vget_low_s8(input_vec));
5641 const int16x8_t second_half = vmovl_s8(vget_high_s8(input_vec));
5726 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
5727 const uint16x8_t first_half = vmovl_u8(vget_low_u8(input_vec));
5728 const uint16x8_t second_half = vmovl_u8(vget_high_u8(input_vec));
5804 const int8x16_t input_vec = vld1q_s8(input_data + i); variable
5805 const int16x8_t first_half = vmovl_s8(vget_low_s8(input_vec));
5806 const int16x8_t second_half = vmovl_s8(vget_high_s8(input_vec));
5881 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2986 std::vector<float> input_vec(n_elements); in TEST_P() local
2987 std::iota(input_vec.begin(), input_vec.end(), 1); in TEST_P()
2988 AddTestTensor("input", p.tensor_dims, tf_type_, input_vec, {}, in TEST_P()
3010 ElementsAreArray(input_vec)); in TEST_P()
3386 std::vector<Input> input_vec{a}; in TEST_P() local
3389 input_vec.push_back(b); in TEST_P()
3391 InputList inputs(input_vec); in TEST_P()
/external/pytorch/aten/src/ATen/cpu/vec/vec256/zarch/
Dvec256_zarch.h1599 auto input_vec = Vectorized<int32_t>::loadu(src_a);
1600 auto output_vec = zvec_convert_to_float(input_vec);
1616 auto input_vec = Vectorized<int64_t>::loadu(src_a);
1617 auto output_vec = zvec_convert_to_float(input_vec);
/external/pytorch/aten/src/ATen/native/cpu/
DSoftMaxKernel.cpp825 Vec input_vec = Vec::loadu(input_data + d * dim_stride); in _vec_softmax() local
826 max_vec = vec::maximum(max_vec, input_vec); in _vec_softmax()
/external/executorch/kernels/optimized/vec/vec256/
Dvec256_int.h298 auto input_vec = _mm256_loadu_si256(reinterpret_cast<const __m256i*>(src + i)); in convert() local
299 auto output_vec = _mm256_cvtepi32_ps(input_vec); in convert()
/external/pytorch/aten/src/ATen/cpu/vec/vec512/
Dvec512_int.h320 auto input_vec = _mm512_loadu_si512(reinterpret_cast<const __m512i*>(src + i)); in convert() local
321 auto output_vec = _mm512_cvtepi32_ps(input_vec); in convert()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc2927 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
2946 shape, input_vec, init_vec, window, compute_tuple)); in TEST_P()
2978 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
2997 shape, input_vec, init_vec, window, compute_tuple)); in TEST_P()
/external/pytorch/aten/src/ATen/cpu/vec/vec256/
Dvec256_int.h288 auto input_vec = _mm256_loadu_si256(reinterpret_cast<const __m256i*>(src + i)); in convert() local
289 auto output_vec = _mm256_cvtepi32_ps(input_vec); in convert()
/external/zstd/lib/common/
Dxxhash.h4250 svuint64_t input_vec = svld1_u64(mask, xinput + offset); \
4252 svuint64_t mixed = sveor_u64_x(mask, secret_vec, input_vec); \
4253 svuint64_t swapped = svtbl_u64(input_vec, kSwap); \