| /external/tensorflow/tensorflow/python/keras/applications/ |
| D | efficientnet.py | 218 input_tensor=None, argument
529 input_tensor=None, argument
555 input_tensor=None, argument
581 input_tensor=None, argument
607 input_tensor=None, argument
633 input_tensor=None, argument
659 input_tensor=None, argument
685 input_tensor=None, argument
711 input_tensor=None, argument
|
| D | densenet.py | 130 input_tensor=None, argument
327 input_tensor=None, argument
340 input_tensor=None, argument
353 input_tensor=None, argument
|
| D | resnet_v2.py | 33 input_tensor=None, argument
64 input_tensor=None, argument
95 input_tensor=None, argument
|
| D | resnet.py | 63 input_tensor=None, argument
444 input_tensor=None, argument
465 input_tensor=None, argument
486 input_tensor=None, argument
|
| D | nasnet.py | 70 input_tensor=None, argument
328 input_tensor=None, argument
405 input_tensor=None, argument
|
| D | mobilenet_v3.py | 165 input_tensor=None, argument
364 input_tensor=None, argument
401 input_tensor=None, argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 584 def reduce_sum(input_tensor, axis=None, keepdims=None, name=None): argument
596 def reduce_prod(input_tensor, axis=None, keepdims=None, name=None): argument
607 def reduce_min(input_tensor, axis=None, keepdims=None, name=None): argument
618 def reduce_max(input_tensor, axis=None, keepdims=None, name=None): argument
629 def reduce_mean(input_tensor, axis=None, keepdims=None, name=None): argument
650 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
660 def reduce_std(input_tensor, axis=None, keepdims=False, name=None): argument
667 def _cast(input_tensor, dtype): argument
672 def reduce_all(input_tensor, axis=None, keepdims=None, name=None): argument
680 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 | 134 auto input_tensor = input.template bit_casted_shaped<T, 3>( in Compute() local
171 const auto& input_tensor = input.flat<T>(); in DequantizeTensor() local
194 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 | 60 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()
106 StatusOr<ElementsAttr> ConvertStringTensor(const Tensor& input_tensor, in ConvertStringTensor()
120 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor()
211 StatusOr<ElementsAttr> ConvertTensorProto(const TensorProto& input_tensor, in ConvertTensorProto()
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 35 def _infer_fft_length_for_rfft(input_tensor, fft_rank): argument
48 def _infer_fft_length_for_irfft(input_tensor, fft_rank): argument
66 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False): argument
117 def _rfft(input_tensor, fft_length=None, name=None): argument
151 def _irfft(input_tensor, fft_length=None, name=None): argument
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 1544 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitAbsNode() local
1628 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitAveragePool2DNode() local
1693 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitCeilNode() local
1731 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitConv2DNode() local
1835 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitDepthwiseConv2DNode() local
1945 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitDepthToSpaceNode() local
2040 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitEluNode() local
2080 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitFullyConnectedNode() local
2234 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitFloorNode() local
2267 const TfLiteTensor& input_tensor = tensors[node->inputs->data[0]]; in VisitHardSwishNode() local
[all …]
|
| /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
555 def all_gather_with_padding(input_tensor, communication_hint, timeout): argument
705 def _pad_util(input_tensor, full_axis_dim): argument
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| 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
|
| D | tflite_text_api.py | 36 def tokenize(self, input_tensor): argument
48 def func(input_tensor): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 2130 def reduce_sum_v1(input_tensor, argument
2207 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): argument
2272 def reduce_sum_with_dims(input_tensor, argument
2285 def reduce_euclidean_norm(input_tensor, axis=None, keepdims=False, name=None): argument
2336 def count_nonzero(input_tensor=None, argument
2476 def reduce_mean_v1(input_tensor, argument
2543 def reduce_mean(input_tensor, axis=None, keepdims=False, name=None): argument
2603 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
2666 def reduce_std(input_tensor, axis=None, keepdims=False, name=None): argument
2717 def reduce_prod(input_tensor, axis=None, keepdims=False, name=None): argument
[all …]
|
| /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
433 const Tensor& input_tensor = ctx->input(0); in DoWork() local
543 const Tensor& input_tensor = ctx->input(0); in DoWork() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Unstack.cpp | 65 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()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | serialization_utils_test.cc | 131 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
|
| /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/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/python/testdata/ |
| D | double_op.py | 30 def double(input_tensor): argument
|
| /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
|