| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | efficientnet.py | 206 input_tensor=None, argument
525 input_tensor=None, argument
551 input_tensor=None, argument
577 input_tensor=None, argument
603 input_tensor=None, argument
629 input_tensor=None, argument
655 input_tensor=None, argument
681 input_tensor=None, argument
707 input_tensor=None, argument
|
| D | densenet.py | 133 input_tensor=None, argument
325 input_tensor=None, argument
338 input_tensor=None, argument
351 input_tensor=None, argument
|
| D | resnet_v2.py | 36 input_tensor=None, argument
67 input_tensor=None, argument
98 input_tensor=None, argument
|
| D | resnet.py | 66 input_tensor=None, argument
458 input_tensor=None, argument
479 input_tensor=None, argument
500 input_tensor=None, argument
|
| D | nasnet.py | 73 input_tensor=None, argument
326 input_tensor=None, argument
403 input_tensor=None, argument
|
| D | mobilenet_v3.py | 158 input_tensor=None, argument
359 input_tensor=None, argument
395 input_tensor=None, argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 567 def reduce_sum(input_tensor, axis=None, keepdims=None, name=None): argument
579 def reduce_prod(input_tensor, axis=None, keepdims=None, name=None): argument
590 def reduce_min(input_tensor, axis=None, keepdims=None, name=None): argument
601 def reduce_max(input_tensor, axis=None, keepdims=None, name=None): argument
612 def reduce_mean(input_tensor, axis=None, keepdims=None, name=None): argument
633 def _cast(input_tensor, dtype): argument
638 def reduce_all(input_tensor, axis=None, keepdims=None, name=None): argument
646 def reduce_any(input_tensor, axis=None, keepdims=None, name=None): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_instance_norm_test.cc | 125 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
|
| D | base64_ops.cc | 34 const Tensor& input_tensor = context->input(0); in Compute() local
59 const Tensor& input_tensor = context->input(0); in Compute() local
|
| D | dequantize_op.cc | 122 auto input_tensor = input.template bit_casted_shaped<T, 3>( in Compute() local
159 const auto& input_tensor = input.flat<T>(); in DequantizeTensor() local
182 const auto& input_tensor = input.flat<T>(); in DequantizeTensor() local
|
| D | regex_full_match_op.cc | 36 const Tensor* input_tensor; in Compute() local
100 const Tensor* input_tensor; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 61 static TensorProto ConvertToProto(const Tensor& input_tensor, in ConvertToProto()
83 StatusOr<ElementsAttr> ConvertFlatTensor(const Tensor& input_tensor, in ConvertFlatTensor()
90 ElementsAttr ConvertBf16Tensor(const Tensor& input_tensor, in ConvertBf16Tensor()
107 StatusOr<ElementsAttr> ConvertStringTensor(const Tensor& input_tensor, in ConvertStringTensor()
121 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor()
171 StatusOr<ElementsAttr> ConvertTensorProto(const TensorProto& input_tensor, in ConvertTensorProto()
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 33 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
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 1047 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitAbsNode() local
1131 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitAveragePool2DNode() local
1196 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitCeilNode() local
1234 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitConv2DNode() local
1329 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitDepthwiseConv2DNode() local
1429 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitDepthToSpaceNode() local
1524 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitEluNode() local
1564 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitFullyConnectedNode() local
1709 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitFloorNode() local
1742 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitHardSwishNode() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 1942 def reduce_sum_v1(input_tensor, argument
2019 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): argument
2084 def reduce_sum_with_dims(input_tensor, argument
2097 def reduce_euclidean_norm(input_tensor, axis=None, keepdims=False, name=None): argument
2148 def count_nonzero(input_tensor=None, argument
2288 def reduce_mean_v1(input_tensor, argument
2355 def reduce_mean(input_tensor, axis=None, keepdims=False, name=None): argument
2415 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
2476 def reduce_std(input_tensor, axis=None, keepdims=False, name=None): argument
2526 def reduce_prod(input_tensor, axis=None, keepdims=False, name=None): argument
[all …]
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| D | tflite_text_api.py | 36 def tokenize(self, input_tensor): argument
48 def func(input_tensor): argument
|
| D | sentencepiece_tokenizer_test.py | 114 def call(self, input_tensor, **kwargs): argument
156 def call(self, input_tensor, **kwargs): argument
198 def call(self, input_tensor, **kwargs): argument
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 76 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
387 const Tensor& input_tensor = ctx->input(0); in DoWork() local
420 const Tensor& input_tensor = ctx->input(0); in DoWork() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 336 input_tensor, argument
366 def _all_gather(self, input_tensor, communication_hint='AUTO', timeout=0): argument
438 input_tensor, argument
698 def _pad_util(input_tensor, full_axis_dim): argument
|
| /external/tensorflow/tensorflow/lite/python/testdata/ |
| D | double_op.py | 30 def double(input_tensor): argument
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | pack_builder.cc | 30 auto& input_tensor = context->tensors[inputs->data[0]]; in GetAxis() local
58 auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 112 auto& input_tensor = input_tensors.back(); in Eval() local
168 TfLiteTensor* input_tensor = &context->tensors[input_tensor_index]; in ResizeOutputTensors() local
265 const auto& input_tensor = context->tensors[tensor_id]; in BuildGraph() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reshape_test.cc | 65 void TestReshapeWithShape(TfLiteTensor* input_tensor, in TestReshapeWithShape()
91 void TestReshapeWithoutShape(TfLiteTensor* input_tensor, in TestReshapeWithoutShape()
124 TfLiteTensor input_tensor = CreateTensor(input_data, input_dims); in TestReshape() local
145 TfLiteTensor input_tensor = CreateQuantizedTensor( in TestReshapeQuantized() local
216 auto input_tensor = tflite::testing::CreateTensor(input_data, input_dims); in TF_LITE_MICRO_TEST() local
335 auto input_tensor = CreateTensor(input_data, input_dims); in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/lite/tools/benchmark/ |
| D | benchmark_test.cc | 104 void CheckInputTensorValue(const TfLiteTensor* input_tensor, in CheckInputTensorValue()
112 void CheckInputTensorValue(const TfLiteTensor* input_tensor, in CheckInputTensorValue()
233 auto input_tensor = benchmark.GetInputTensor(0); in TEST() local
310 auto input_tensor = benchmark.GetInputTensor(0); in TEST() local
405 auto input_tensor = interpreter->tensor(inputs[0]); in TEST() local
|
| /external/tensorflow/tensorflow/security/fuzzing/op_fuzzing/ |
| D | identity_fuzz.cc | 26 Tensor input_tensor(tensorflow::DT_INT8, in FuzzImpl() local
|