Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
Dregex_replace_op_test.py50 input_vector = constant_op.constant(values, dtypes.string)
51 stripped = op(input_vector, "^(a:|b:)", "", replace_global=False)
59 input_vector = constant_op.constant(values, dtypes.string)
60 stripped = op(input_vector, "a.*a", "(\\0)")
67 input_vector = constant_op.constant(values, dtypes.string)
68 stripped = op(input_vector, "", "x")
75 input_vector = constant_op.constant(values, dtypes.string)
77 replace = op(input_vector, invalid_pattern, "x")
85 input_vector = constant_op.constant(values, dtypes.string)
86 stripped = op(input_vector, "ab", "abc", True)
[all …]
Dunicode_script_op_test.py37 input_vector = constant_op.constant(inputs, dtypes.int32)
38 outputs = string_ops.unicode_script(input_vector).eval()
52 input_vector = constant_op.constant(inputs, dtypes.int32)
53 outputs = string_ops.unicode_script(input_vector).eval()
/external/tensorflow/tensorflow/python/
Dpywrap_dtensor_device.cc228 std::vector<TFE_TensorHandle*> input_vector; in PYBIND11_MODULE() local
229 input_vector.resize(len); in PYBIND11_MODULE()
231 input_vector[i] = EagerTensor_Handle(py_eager_tensor_handles[i].get()); in PYBIND11_MODULE()
234 auto size = input_vector.size() / 3; in PYBIND11_MODULE()
237 /*num_inputs=*/input_vector.size() / 3, in PYBIND11_MODULE()
239 std::vector<TFE_TensorHandle*>(input_vector.begin(), in PYBIND11_MODULE()
240 input_vector.begin() + size) in PYBIND11_MODULE()
243 std::vector<TFE_TensorHandle*>(input_vector.begin() + size, in PYBIND11_MODULE()
244 input_vector.begin() + 2 * size) in PYBIND11_MODULE()
247 std::vector<TFE_TensorHandle*>(input_vector.begin() + 2 * size, in PYBIND11_MODULE()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dpropagator_debug_utils.cc43 void DumpPendingNodeState(const NodeItem& node_item, const Entry* input_vector, in DumpPendingNodeState() argument
49 const Entry& input = input_vector[input_base + i]; in DumpPendingNodeState()
62 const Entry& input = input_vector[input_base + i]; in DumpPendingNodeState()
75 void DumpActiveNodeState(const NodeItem& node_item, const Entry* input_vector) { in DumpActiveNodeState() argument
79 const Entry& input = input_vector[input_base + i]; in DumpActiveNodeState()
Dpropagator_debug_utils.h29 void DumpPendingNodeState(const NodeItem& node_item, const Entry* input_vector,
34 void DumpActiveNodeState(const NodeItem& node_item, const Entry* input_vector);
/external/armnn/src/backends/reference/workloads/
DLstmUtils.cpp40 void MeanStddevNormalization(armnn::Decoder<float>& input_vector, in MeanStddevNormalization() argument
50 sum += input_vector.Get(); in MeanStddevNormalization()
51 sum_sq += input_vector.Get() * input_vector.Get(); in MeanStddevNormalization()
52 ++input_vector; in MeanStddevNormalization()
54 input_vector -= v_size; in MeanStddevNormalization()
66 output_vector.Set((input_vector.Get() - mean) * stddev_inv); in MeanStddevNormalization()
68 ++input_vector; in MeanStddevNormalization()
73 input_vector -= v_size * n_batch; in MeanStddevNormalization()
DLstmUtils.hpp23 void MeanStddevNormalization(armnn::Decoder<float>& input_vector,
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreverse_test.cc82 std::vector<float> input_vector( in TEST_P() local
84 std::iota(input_vector.begin(), input_vector.end(), 0.0); in TEST_P()
85 auto r1_literal = LiteralUtil::CreateR1<float>(input_vector); in TEST_P()
Dreduce_window_test.cc507 std::vector<float> input_vector(128 * 9, 1); in XLA_TEST_P() local
509 LiteralUtil::CreateR1<float>(input_vector), &builder_); in XLA_TEST_P()
518 std::vector<float> input_vector{ in XLA_TEST_P() local
528 LiteralUtil::CreateR1<float>(input_vector), &builder_); in XLA_TEST_P()
535 std::vector<float> input_vector{ in XLA_TEST_P() local
545 LiteralUtil::CreateR1<float>(input_vector), &builder_); in XLA_TEST_P()
1521 std::vector<float> input_vector(param.base_bounds[0]); in XLA_TEST_P() local
1522 std::iota(std::begin(input_vector), std::end(input_vector), 0); in XLA_TEST_P()
1524 LiteralUtil::CreateR1(absl::Span<const float>(input_vector)); in XLA_TEST_P()
1552 /*operand=*/absl::Span<const float>(input_vector), in XLA_TEST_P()
/external/tensorflow/tensorflow/lite/kernels/
Dwhile_test.cc93 const std::vector<int> input_vector(1000000, 1); in TEST_F() local
95 input_vector); in TEST_F()
152 std::vector<int> input_vector(1000000, 0); in TEST_F() local
153 input_vector[0] = 5; in TEST_F()
154 input_vector[1] = 7; in TEST_F()
155 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), input_vector); in TEST_F()
Dif_test.cc75 const std::vector<int> input_vector(kNumLargeTensors, 1); in TEST_F() local
77 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), input_vector); in TEST_F()
101 const std::vector<int> input_vector(kNumLargeTensors, 1); in TEST_F() local
103 FillIntTensor(interpreter_->tensor(interpreter_->inputs()[1]), input_vector); in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dportable_tensor_utils.h294 void ReductionSumVector(const float* input_vector, float* output_vector, in ReductionSumVector() argument
296 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
300 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
302 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
306 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
308 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
312 void MeanStddevNormalization(const float* input_vector, float* output_vector, in MeanStddevNormalization() argument
314 PortableMeanStddevNormalization(input_vector, output_vector, v_size, n_batch); in MeanStddevNormalization()
Dportable_tensor_utils_impl.h213 void PortableReductionSumVector(const INPUT* input_vector, in PortableReductionSumVector() argument
219 result += input_vector[r]; in PortableReductionSumVector()
222 input_vector += reduction_size; in PortableReductionSumVector()
227 void PortableMeanStddevNormalization(const float* __restrict__ input_vector,
Dportable_tensor_utils.cc753 void PortableMeanStddevNormalization(const float* __restrict__ input_vector, in PortableMeanStddevNormalization() argument
759 sum += input_vector[i]; in PortableMeanStddevNormalization()
764 const float diff = input_vector[i] - mean; in PortableMeanStddevNormalization()
772 output_vector[i] = (input_vector[i] - mean) * stddev_inv; in PortableMeanStddevNormalization()
774 input_vector += v_size; in PortableMeanStddevNormalization()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.h305 void ReductionSumVector(const float* input_vector, float* output_vector, in ReductionSumVector() argument
307 NEON_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
311 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
313 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
317 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
319 SSE_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
323 void MeanStddevNormalization(const float* __restrict__ input_vector, in MeanStddevNormalization() argument
326 PortableMeanStddevNormalization(input_vector, output_vector, v_size, n_batch); in MeanStddevNormalization()
Dneon_tensor_utils.h290 void ReductionSumVector(const float* input_vector, float* output_vector, in ReductionSumVector() argument
292 NEON_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
296 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
298 PortableReductionSumVector(input_vector, output_vector, output_size, in ReductionSumVector()
302 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in ReductionSumVector() argument
304 NEON_OR_PORTABLE(ReductionSumVector, input_vector, output_vector, output_size, in ReductionSumVector()
308 void MeanStddevNormalization(const float* __restrict__ input_vector, in MeanStddevNormalization() argument
311 NEON_OR_PORTABLE(MeanStddevNormalization, input_vector, output_vector, v_size, in MeanStddevNormalization()
Dneon_tensor_utils_impl.h176 void NeonReductionSumVector(const float* input_vector, float* output_vector,
179 void NeonReductionSumVector(const int8_t* input_vector, int32_t* output_vector,
187 void NeonMeanStddevNormalization(const float* __restrict__ input_vector,
Dneon_tensor_utils.cc2565 void NeonReductionSumVector(const float* input_vector, float* output_vector, in NeonReductionSumVector() argument
2576 float32x4_t v1_f32x4 = vld1q_f32(input_vector + r); in NeonReductionSumVector()
2582 sum += input_vector[r]; in NeonReductionSumVector()
2585 input_vector += reduction_size; in NeonReductionSumVector()
2589 void NeonReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in NeonReductionSumVector() argument
2599 const int8x16_t s2_8x16 = vld1q_s8(input_vector + r); in NeonReductionSumVector()
2603 const int8x8_t s2_8x8 = vld1_s8(input_vector + r); in NeonReductionSumVector()
2609 sum += input_vector[r]; in NeonReductionSumVector()
2612 input_vector += reduction_size; in NeonReductionSumVector()
2676 void NeonMeanStddevNormalization(const float* __restrict__ input_vector, in NeonMeanStddevNormalization() argument
[all …]
Dsse_tensor_utils_impl.h79 void SseReductionSumVector(const int8_t* input_vector, int32_t* output_vector,
Dsse_tensor_utils.cc620 void SseReductionSumVector(const int8_t* input_vector, int32_t* output_vector, in SseReductionSumVector() argument
624 const int8_t* __restrict__ row_ptr = input_vector + row * reduction_size; in SseReductionSumVector()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h75 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleNode() argument
85 tuple_node.tuples(i), input_tensor_list, input_vector, in ParseTupleNode()
91 if (input_index < 0 || input_index >= input_vector->size()) { in ParseTupleNode()
94 input_vector->size(), " inputs."); in ParseTupleNode()
98 input_vector->at(input_index); in ParseTupleNode()
127 std::vector<XRTTupleAllocation::ExpandedTupleInput>* input_vector, in ParseTupleTree() argument
135 input_vector, &tuple_tree_shape, rm)); in ParseTupleTree()
153 *element = input_vector->at(input_index); in ParseTupleTree()
444 std::vector<XRTTupleAllocation::ExpandedTupleInput> input_vector( in Compute()
456 tuple_proto, arg_list, &input_vector, in Compute()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dportable_tensor_utils.h456 void ReductionSumVector(const float* input_vector, float* output_vector,
460 void ReductionSumVector(const int32_t* input_vector, int32_t* output_vector,
464 void ReductionSumVector(const int8_t* input_vector, int32_t* output_vector,
468 void MeanStddevNormalization(const float* input_vector, float* output_vector,
/external/webrtc/modules/audio_coding/neteq/
Dmerge.cc62 AudioMultiVector input_vector(num_channels_); in Process() local
63 input_vector.PushBackInterleaved( in Process()
65 size_t input_length_per_channel = input_vector.Size(); in Process()
75 input_vector[channel].CopyTo(input_length_per_channel, 0, in Process()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp142 std::vector<const ICLTensor *> input_vector; in configure() local
143 input_vector.emplace_back(input); in configure()
144 input_vector.emplace_back(output_state_in); in configure()
148 _concat_inputs.configure(compile_context, input_vector, &_input, Window::DimX); in configure()
360 std::vector<const ITensorInfo *> input_vector; in validate() local
361 input_vector.emplace_back(input); in validate()
362 input_vector.emplace_back(output_state_in); in validate()
364 …ARM_COMPUTE_RETURN_ON_ERROR(CLConcatenateLayer::validate(input_vector, &input_concatenated, Window… in validate()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayerQuantized.cpp118 std::vector<const ITensor *> input_vector{ input, output_state_in }; in configure() local
121 _concat_inputs.configure(input_vector, &_input, Window::DimX); in configure()
327 std::vector<const ITensorInfo *> input_vector; in validate() local
328 input_vector.emplace_back(input); in validate()
329 input_vector.emplace_back(output_state_in); in validate()
331 …ARM_COMPUTE_RETURN_ON_ERROR(NEConcatenateLayer::validate(input_vector, &input_concatenated, Window… in validate()