Home
last modified time | relevance | path

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

1234567

/external/tensorflow/tensorflow/python/keras/applications/
Defficientnet.py155 input_tensor=None, argument
459 input_tensor=None, argument
483 input_tensor=None, argument
507 input_tensor=None, argument
531 input_tensor=None, argument
555 input_tensor=None, argument
579 input_tensor=None, argument
603 input_tensor=None, argument
627 input_tensor=None, argument
Dresnet_v2.py30 input_tensor=None, argument
48 input_tensor=None, argument
66 input_tensor=None, argument
Ddensenet.py131 input_tensor=None, argument
306 input_tensor=None, argument
319 input_tensor=None, argument
332 input_tensor=None, argument
Dresnet.py60 input_tensor=None, argument
439 input_tensor=None, argument
460 input_tensor=None, argument
481 input_tensor=None, argument
Dnasnet.py72 input_tensor=None, argument
313 input_tensor=None, argument
381 input_tensor=None, argument
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_math_ops.py546 def reduce_sum(input_tensor, axis=None, keepdims=None, name=None): argument
557 def reduce_prod(input_tensor, axis=None, keepdims=None, name=None): argument
568 def reduce_min(input_tensor, axis=None, keepdims=None, name=None): argument
579 def reduce_max(input_tensor, axis=None, keepdims=None, name=None): argument
590 def reduce_mean(input_tensor, axis=None, keepdims=None, name=None): argument
609 def _cast(input_tensor, dtype): argument
614 def reduce_all(input_tensor, axis=None, keepdims=None, name=None): argument
622 def reduce_any(input_tensor, axis=None, keepdims=None, name=None): argument
/external/tensorflow/tensorflow/core/kernels/
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
Dbase64_ops.cc34 const Tensor& input_tensor = context->input(0); in Compute() local
59 const Tensor& input_tensor = context->input(0); in Compute() local
Dregex_full_match_op.cc32 const Tensor* input_tensor; in Compute() local
72 const Tensor* input_tensor; in Compute() local
Dstring_to_hash_bucket_op.h37 const Tensor* input_tensor; in Compute() local
79 const Tensor* input_tensor; in Compute() local
Ddequantize_op.cc119 auto input_tensor = input.template bit_casted_shaped<T, 3>( in Compute() local
154 const auto& input_tensor = input.flat<T>(); in DequantizeTensor() local
177 const auto& input_tensor = input.flat<T>(); in DequantizeTensor() local
Dguarantee_const_op.cc32 const Tensor& input_tensor = ctx->input(0); in Compute() local
Dstring_strip_op.cc34 const Tensor* input_tensor; in Compute() local
Dstring_to_number_op.cc41 const Tensor* input_tensor; in Compute() local
Dunicode_script_op.cc27 const Tensor* input_tensor; in Compute() local
/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py32 def _infer_fft_length_for_rfft(input_tensor, fft_rank): argument
45 def _infer_fft_length_for_irfft(input_tensor, fft_rank): argument
63 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False): argument
114 def _rfft(input_tensor, fft_length=None, name=None): argument
144 def _irfft(input_tensor, fft_length=None, name=None): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor.cc57 static TensorProto ConvertToProto(const Tensor& input_tensor, in ConvertToProto()
79 StatusOr<ElementsAttr> ConvertFlatTensor(const Tensor& input_tensor, in ConvertFlatTensor()
86 StatusOr<ElementsAttr> ConvertBF16Tensor(const Tensor& input_tensor, in ConvertBF16Tensor()
98 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor()
135 StatusOr<ElementsAttr> ConvertTensorProto(const TensorProto& input_tensor, in ConvertTensorProto()
/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py1608 def reduce_sum_v1(input_tensor, argument
1663 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): argument
1706 def reduce_sum_with_dims(input_tensor, argument
1718 def reduce_euclidean_norm(input_tensor, axis=None, keepdims=False, name=None): argument
1765 def count_nonzero(input_tensor=None, argument
1903 def reduce_mean_v1(input_tensor, argument
1970 def reduce_mean(input_tensor, axis=None, keepdims=False, name=None): argument
2029 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
2077 def reduce_std(input_tensor, axis=None, keepdims=False, name=None): argument
2124 def reduce_prod(input_tensor, axis=None, keepdims=False, name=None): argument
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dreshape_test.cc34 void TestReshapeImpl(TfLiteTensor* input_tensor, TfLiteTensor* shape_tensor, in TestReshapeImpl()
113 TfLiteTensor input_tensor = CreateTensor<T, tensor_input_type>( in TestReshape() local
191 auto input_tensor = CreateFloatTensor(input_data, input_dims, "input_tensor"); in TF_LITE_MICRO_TEST() local
255 auto input_tensor = CreateFloatTensor(input_data, input_dims, "input_tensor"); in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/examples/adding_an_op/
Dcuda_op_kernel.cc39 const Tensor& input_tensor = context->input(0); in Compute() local
/external/tensorflow/tensorflow/core/kernels/fuzzing/
Didentity_fuzz.cc26 Tensor input_tensor(tensorflow::DT_INT8, in FuzzImpl() local
Dstring_split_fuzz.cc33 Tensor input_tensor(tensorflow::DT_STRING, TensorShape({})); in FuzzImpl() local
/external/tensorflow/tensorflow/tools/ci_build/builds/user_ops/
Dcuda_op_kernel.cc39 const Tensor& input_tensor = context->input(0); in Compute() local
Dzero_out_op_kernel_1.cc42 const Tensor& input_tensor = context->input(0); in Compute() local
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Ddense_to_sparse_batch_test.py95 def dataset_fn(input_tensor): argument

1234567