Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dfeature_column.py251 input_tensor, argument
258 def _deep_embedding_lookup_arguments(self, input_tensor): argument
266 def _wide_embedding_lookup_arguments(self, input_tensor): argument
272 def _to_dense_tensor(self, input_tensor): argument
414 def id_tensor(self, input_tensor): argument
419 def weight_tensor(self, input_tensor): argument
425 input_tensor, argument
434 def _wide_embedding_lookup_arguments(self, input_tensor): argument
442 def _get_input_sparse_tensor(self, input_tensor): argument
471 def _do_transform(self, input_tensor): argument
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_math_ops.py501 def reduce_sum(input_tensor, axis=None, keepdims=None, name=None): argument
508 def reduce_prod(input_tensor, axis=None, keepdims=None, name=None): argument
515 def reduce_min(input_tensor, axis=None, keepdims=None, name=None): argument
522 def reduce_max(input_tensor, axis=None, keepdims=None, name=None): argument
529 def reduce_mean(input_tensor, axis=None, keepdims=None, name=None): argument
547 def _cast(input_tensor, dtype): argument
552 def reduce_all(input_tensor, axis=None, keepdims=None, name=None): argument
560 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
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
/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py31 def _infer_fft_length_for_rfft(input_tensor, fft_rank): argument
44 def _infer_fft_length_for_irfft(input_tensor, fft_rank): argument
62 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False): argument
113 def _rfft(input_tensor, fft_length=None, name=None): argument
132 def _irfft(input_tensor, fft_length=None, name=None): argument
/external/tensorflow/tensorflow/contrib/text/python/ops/
Dskip_gram_ops.py41 def skip_gram_sample(input_tensor, argument
210 def skip_gram_sample_with_text_vocab(input_tensor, argument
390 def _filter_input(input_tensor, vocab_freq_table, vocab_min_count, argument
/external/tensorflow/tensorflow/python/ops/
Dmath_ops.py1315 def reduce_sum_v1(input_tensor, argument
1369 def reduce_sum(input_tensor, axis=None, keepdims=False, name=None): argument
1416 def reduce_euclidean_norm(input_tensor, axis=None, keepdims=False, name=None): argument
1461 def count_nonzero(input_tensor=None, argument
1600 def reduce_mean_v1(input_tensor, argument
1664 def reduce_mean(input_tensor, axis=None, keepdims=False, name=None): argument
1721 def reduce_variance(input_tensor, axis=None, keepdims=False, name=None): argument
1769 def reduce_std(input_tensor, axis=None, keepdims=False, name=None): argument
1816 def reduce_prod(input_tensor, axis=None, keepdims=False, name=None): argument
1853 def reduce_prod_v1(input_tensor, argument
[all …]
/external/tensorflow/tensorflow/contrib/layers/python/ops/
Dbucketization_op.py23 def bucketize(input_tensor, boundaries, name=None): argument
/external/tensorflow/tensorflow/contrib/session_bundle/
Dexporter.py77 def regression_signature(input_tensor, output_tensor): argument
95 def classification_signature(input_tensor, argument
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc127 TfLiteStatus SymmetricPerChannelBiasQuantize(const TensorT* input_tensor, in SymmetricPerChannelBiasQuantize()
251 const TensorT* input_tensor = subgraph_->tensors[input_tensor_idx].get(); in QuantizeOpWithBias() local
279 auto input_tensor = subgraph_->tensors[op->inputs[0]].get(); in PropagateMinMaxForAvgAndMaxPool() local
336 auto input_tensor = subgraph_->tensors[op->inputs[input_idx]].get(); in AsymmetricQuantizeInputsAndOutputs() local
/external/tensorflow/tensorflow/contrib/kernel_methods/python/mappers/
Ddense_kernel_mapper.py40 def map(self, input_tensor): argument
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api_test.cc52 TFL_Tensor* input_tensor = TFL_InterpreterGetInputTensor(interpreter, 0); in TEST() local
114 TFL_Tensor* input_tensor = TFL_InterpreterGetInputTensor(interpreter, 0); in TEST() local
/external/tensorflow/tensorflow/examples/adding_an_op/
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/tools/ci_build/builds/user_ops/
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/contrib/quantize/python/
Dfold_batch_norms.py446 def _CloneWithNewOperands(layer_op, input_tensor, weight_tensor, argument
901 def _AssertConvShapes(self, op_name, input_tensor, weights): argument
919 def _AssertFCShapes(self, op_name, weights, input_tensor): argument
1001 def __init__(self, layer_op, bn_op, output_tensor, input_tensor, argument
1035 def input_tensor(self): member in _BatchNormMatch
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Ddense_to_sparse_batch_test.py91 def dataset_fn(input_tensor): argument
/external/tensorflow/tensorflow/contrib/gan/python/eval/python/
Deval_utils_impl.py34 def image_grid(input_tensor, grid_shape, image_shape=(32, 32), num_channels=3): argument

123456