Searched refs:tensor_spec_value (Results 1 – 6 of 6) sorted by relevance
401 def do_encode(self, tensor_spec_value, encode_fn): argument403 encoded_tensor_spec.tensor_spec_value.CopyFrom(405 shape=encode_fn(tensor_spec_value.shape).tensor_shape_value,406 dtype=encode_fn(tensor_spec_value.dtype).tensor_dtype_value,407 name=tensor_spec_value.name))414 name = value.tensor_spec_value.name418 tensor_shape_value=value.tensor_spec_value.shape)),421 tensor_dtype_value=value.tensor_spec_value.dtype)),
167 expected_tensor_spec = expected_list.values.add().tensor_spec_value183 expected_tensor_spec = expected_list.values.add().tensor_spec_value
132 PartialTensorShape shape = arg.tensor_spec_value().shape(); in ImportModule()133 DataType dtype = arg.tensor_spec_value().dtype(); in ImportModule()140 saved_concrete_function.output_signature().tensor_spec_value(); in ImportModule()
214 result[key] = &value.tensor_spec_value(); in SignatureDefArgsFromInputs()248 result[key] = &value.tensor_spec_value(); in SignatureDefReturnsFromOutputs()
61 TensorSpecProto tensor_spec_value = 33; field
318 flattened_specs->push_back(&signature.tensor_spec_value()); in FlattenSignature()