Home
last modified time | relevance | path

Searched refs:output_signature (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dpartially_revived_objects.cc226 Status SignatureDefReturnsFromOutputs(const StructuredValue& output_signature, in SignatureDefReturnsFromOutputs() argument
228 if (!output_signature.has_dict_value()) { in SignatureDefReturnsFromOutputs()
232 output_signature.DebugString()); in SignatureDefReturnsFromOutputs()
235 const DictValue& output_dict = output_signature.dict_value(); in SignatureDefReturnsFromOutputs()
271 saved_concrete_function.output_signature(), &rets)); in LoadSignatureDefFunctionMetadata()
/external/tensorflow/tensorflow/python/keras/layers/
Dcore_test.py164 output_signature = l.compute_output_signature([
167 self.assertAllEqual((10, 20), output_signature.shape)
168 self.assertAllEqual(dtypes.float64, output_signature.dtype)
518 output_signature = layer.compute_output_signature(
520 self.assertEqual(output_signature.dtype, dtypes.float16)
521 self.assertEqual(output_signature.shape, (2, 5))
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py737 output_signature=None): argument
809 if output_signature is not None:
818 for _ in nest.flatten(output_signature)):
826 if output_signature is None:
834 output_signature = nest.map_structure_up_to(output_types,
839 for x in nest.flatten(output_signature)
842 output_signature, [x.dtype for x in nest.flatten(output_signature)])
844 output_signature, [x.shape for x in nest.flatten(output_signature)])
955 flat_output_types = structure.get_flat_tensor_types(output_signature)
964 values = structure.normalize_element(values, output_signature)
[all …]
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsaved_model_utils.cc171 const StructuredValue& output_signature = in ValidateSavedFunctionCompatibleWithFunctionDef() local
172 saved_concrete_function.output_signature(); in ValidateSavedFunctionCompatibleWithFunctionDef()
174 TF_RETURN_IF_ERROR(FlattenSignature(output_signature, &output_specs)); in ValidateSavedFunctionCompatibleWithFunctionDef()
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py80 concrete_function_proto.output_signature.CopyFrom(
/external/tensorflow/tensorflow/python/keras/utils/
Dkpl_test_utils.py109 output_signature={
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dfrom_generator_test.py464 output_signature=ragged_tensor.RaggedTensorSpec(
484 output_signature=sparse_tensor.SparseTensorSpec([3, 4], dtypes.int64))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.-dataset.pbtxt51 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.-fixed-length-record-dataset.pbtxt53 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.-text-line-dataset.pbtxt53 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.experimental.-csv-dataset.pbtxt53 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.experimental.-random-dataset.pbtxt53 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.experimental.-sql-dataset.pbtxt53 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.-t-f-record-dataset.pbtxt52 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_test.py157 output_signature = layer.compute_output_signature(
161 self.assertEqual(output_signature.dtype, dtypes.float64)
162 self.assertEqual(output_signature.shape.as_list(), [None, 10])
1646 output_signature = layer.compute_output_signature(
1648 self.assertEqual(output_signature.shape, ())
1649 self.assertEqual(output_signature.dtype, 'float64')
/external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/
Dsaved_model.pbtxt9834 output_signature {
9888 output_signature {
9943 output_signature {
9989 output_signature {
10044 output_signature {
10099 output_signature {
10163 output_signature {
10227 output_signature {
10291 output_signature {
10355 output_signature {
[all …]
/external/tensorflow/tensorflow/python/keras/integration_test/
Dtpu_strategy_test.py136 output_signature={
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.-dataset.pbtxt68 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.-t-f-record-dataset.pbtxt70 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.experimental.-random-dataset.pbtxt70 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.experimental.-sql-dataset.pbtxt70 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.experimental.-csv-dataset.pbtxt70 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.-text-line-dataset.pbtxt70 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
Dtensorflow.data.-fixed-length-record-dataset.pbtxt70 …gs=[\'generator\', \'output_types\', \'output_shapes\', \'args\', \'output_signature\'], varargs=N…
/external/tensorflow/tensorflow/python/keras/distribute/
Dparameter_server_training_test.py147 output_signature={

12