Home
last modified time | relevance | path

Searched defs:input_tensor (Results 1 – 25 of 280) sorted by relevance

12345678910>>...12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc1767 TfLiteContext* context, const TfLiteTensor& input_tensor, in CheckTensorsDimensionMatch()
1811 TfLiteContext* context, const TfLiteTensor& input_tensor, in CheckTensorsInputOutputScale()
2153 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitAbsNode() local
2249 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitAveragePool2DNode() local
2315 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitCeilNode() local
2361 const TfLiteTensor& input_tensor = tensors[input_idx]; in VisitSplitNode() local
2477 const TfLiteTensor& input_tensor = tensors[node->inputs->data[i]]; in VisitConcatenationNode() local
2492 const TfLiteTensor& input_tensor = tensors[node->inputs->data[i]]; in VisitConcatenationNode() local
2558 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitConv2DNode() local
2667 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitDepthwiseConv2DNode() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dmath_ops.py2174 def reduce_sum_v1(input_tensor, argument
2251 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): argument
2316 def reduce_sum_with_dims(input_tensor, argument
2329 def reduce_euclidean_norm(input_tensor, axis=None, keepdims=False, name=None): argument
2380 def count_nonzero(input_tensor=None, argument
2520 def reduce_mean_v1(input_tensor, argument
2587 def reduce_mean(input_tensor, axis=None, keepdims=False, name=None): argument
2647 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
2710 def reduce_std(input_tensor, axis=None, keepdims=False, name=None): argument
2761 def reduce_prod(input_tensor, axis=None, keepdims=False, name=None): argument
[all …]
H A Dimage_grad_test_base.py305 def func(input_tensor, align_corners=align_corners): argument
324 def func(input_tensor, align_corners=align_corners): argument
372 def scale_trans(input_tensor, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dconvert_tensor.cc60 static TensorProto ConvertToProto(const Tensor& input_tensor, in ConvertToProto()
82 StatusOr<ElementsAttr> ConvertFlatTensor(const Tensor& input_tensor, in ConvertFlatTensor()
89 ElementsAttr ConvertBf16Tensor(const Tensor& input_tensor, in ConvertBf16Tensor()
102 StatusOr<ElementsAttr> ConvertStringTensor(const Tensor& input_tensor, in ConvertStringTensor()
116 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor()
204 StatusOr<ElementsAttr> ConvertTensorProto(const TensorProto& input_tensor, in ConvertTensorProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dconvert_tensor.cc57 static TensorProto ConvertToProto(const Tensor& input_tensor, in ConvertToProto()
79 tensorflow::StatusOr<ElementsAttr> ConvertFlatTensor(const Tensor& input_tensor, in ConvertFlatTensor()
86 ElementsAttr ConvertBf16Tensor(const Tensor& input_tensor, in ConvertBf16Tensor()
100 const Tensor& input_tensor, ShapedType type) { in ConvertStringTensor()
113 tensorflow::StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor()
203 const TensorProto& input_tensor, Builder builder) { in ConvertTensorProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_instance_norm_test.cc125 Tensor input_tensor(DT_QUINT8, {1, 4, 4, 32}); in TestBasic() local
134 Tensor input_tensor(DT_QUINT8, {1, 4, 4, 32}); in TestZeroInput() local
144 Tensor input_tensor(DT_QUINT8, {1, 1, 2, 16}); in TestMaxInput() local
155 Tensor input_tensor(DT_QUINT8, {1, 4, 4, 32}); in TestOutputRangeGiven() local
163 Tensor input_tensor(DT_QUINT8, {1, 4, 4, 32}); in TestClamp() local
H A Ddequantize_op.cc139 auto input_tensor = input.template bit_casted_shaped<T, 3>( in Compute() local
176 const auto& input_tensor = input.flat<T>(); in DequantizeTensor() local
199 const auto& input_tensor = input.flat<T>(); in DequantizeTensor() local
H A Dbase64_ops.cc34 const Tensor& input_tensor = context->input(0); in Compute() local
59 const Tensor& input_tensor = context->input(0); in Compute() local
H A Dregex_full_match_op.cc36 const Tensor* input_tensor; in Compute() local
100 const Tensor* input_tensor; in Compute() local
H A Dunicode_ops.cc275 const Tensor* input_tensor; in Compute() local
405 const Tensor* input_tensor; in Compute() local
531 const Tensor& input_tensor = context->input(0); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/signal/
H A Dfft_ops.py33 def _infer_fft_length_for_rfft(input_tensor, fft_rank): argument
46 def _infer_fft_length_for_irfft(input_tensor, fft_rank): argument
64 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False): argument
115 def _rfft(input_tensor, fft_length=None, name=None): argument
149 def _irfft(input_tensor, fft_length=None, name=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dinfeed_ops.cc76 const Tensor& input_tensor, in TransposeTensor()
196 const Tensor& input_tensor, in AutoTransposeAndLinearize()
251 const Tensor& input_tensor = ctx->input(0); in Compute() local
329 const Tensor& input_tensor = values[i]; in Compute() local
435 const Tensor& input_tensor = ctx->input(0); in DoWork() local
550 const Tensor& input_tensor = ctx->input(0); in DoWork() local
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_executor.cpp96 auto input_tensor = subgraph->tensors[input_index]; in patch_conv2d() local
182 auto input_tensor = subgraph->tensors[input_index]; in patch_quant_for_add() local
214 auto input_tensor = subgraph->tensors[input_index]; in add_generate_model() local
285 TfLiteTensor *input_tensor = TfLiteInterpreterGetInputTensor(interpreter, i); in run_model() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/benchmark/
H A Dbenchmark_test.cc107 void CheckInputTensorValue(const TfLiteTensor* input_tensor, in CheckInputTensorValue()
115 void CheckInputTensorValue(const TfLiteTensor* input_tensor, in CheckInputTensorValue()
260 auto input_tensor = benchmark.GetInputTensor(0); in TEST() local
337 auto input_tensor = benchmark.GetInputTensor(0); in TEST() local
432 auto input_tensor = interpreter->tensor(inputs[0]); in TEST() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/python/
H A Dsentencepiece_tokenizer_test.py114 def call(self, input_tensor, **kwargs): argument
156 def call(self, input_tensor, **kwargs): argument
198 def call(self, input_tensor, **kwargs): argument
H A Dtflite_text_api.py36 def tokenize(self, input_tensor): argument
48 def func(input_tensor): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dc_test.c102 TfLiteTensor* input_tensor = in TestInferenceUsingSignature() local
177 TfLiteTensor* input_tensor = in TestRepeatResizeInputTensor() local
257 TfLiteTensor* input_tensor = TfLiteInterpreterGetInputTensor(interpreter, 0); in TestInferenceUsingInterpreter() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dserialization_utils_test.cc131 Tensor input_tensor(DT_FLOAT, {1}); in TEST() local
168 Tensor input_tensor(DT_FLOAT, {1}); in TEST() local
209 Tensor input_tensor(DT_FLOAT, {1}); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dhexagon_delegate_kernel.cc114 auto& input_tensor = input_tensors.back(); in Eval() local
170 TfLiteTensor* input_tensor = &context->tensors[input_tensor_index]; in ResizeOutputTensors() local
269 const auto& input_tensor = context->tensors[tensor_id]; in BuildGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dquantize_nodes_test.cc323 Tensor input_tensor(DT_FLOAT, TensorShape({input_batch_count, input_height, in TestQuantizeConv2D() local
348 Tensor input_tensor(DT_FLOAT, TensorShape({1, 1, 2, 6})); in TestQuantizeBiasAdd() local
674 Tensor input_tensor(DT_FLOAT, {1, 128, 128, 3}); in TestQuantizeResizeBilinear() local
805 Tensor input_tensor(DT_FLOAT, {}); in TestQuantizePlaceholders() local
879 Tensor input_tensor(DT_FLOAT, TensorShape({1, 1, 2, 6})); in TestConvertFakeQuantsToRequantize() local
923 Tensor input_tensor(DT_QUINT8, TensorShape({1, 1, 2, 6})); in TestMergeAdjacentRequantizes() local
1031 Tensor input_tensor(DT_FLOAT, TensorShape({1, 1, 2, 6})); in TestConvertFakeQuantsEndToEnd() local
1080 Tensor input_tensor(DT_FLOAT, TensorShape({1, 1, 2, 6})); in TestHoistFakeQuants() local
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/
H A Dcluster_coordinator_test.py1062 def replica_fn(input_tensor): argument
1139 def worker_fn(input_tensor): argument
1141 def replica_fn(input_tensor): argument
1176 def worker_fn(input_tensor): argument
1178 def replica_fn(input_tensor): argument
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py656 def replace(self, input_tensor, output_tensor) -> None: argument
704 def replace(self, input_tensor, output_tensor) -> None: argument
745 def replace(self, input_tensor, output_tensor) -> None: argument
791 def replace(self, input_tensor, output_tensor) -> None: argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Dmath_utils.h68 void AffineQuantize(const ConstTensorTin& input_tensor, float inv_scale, in AffineQuantize()
84 void AffineDequantize(const ConstTensorTin& input_tensor, float scale, in AffineDequantize()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DUnstack.cpp65 SimpleTensor<T> get_slice(const SimpleTensor<T> &input_tensor, size_t axis, size_t slice) in get_slice()
87 std::vector<SimpleTensor<T>> unstack(const SimpleTensor<T> &input_tensor, std::vector<SimpleTensor<… in unstack()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dpack_builder.cc30 auto& input_tensor = context->tensors[inputs->data[0]]; in GetAxis() local
58 auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local

12345678910>>...12