Searched refs:tensor_specs (Results 1 – 10 of 10) sorted by relevance
358 tensor_specs = []368 tensor_specs.extend(372 return tensor_specs446 tensor_specs = []451 tensor_specs = output_tensor_specs[456 tensor_specs.append(output_tensor_specs[output_tensor_spec_index])458 grouped_output_tensor_specs.append(tensor_specs)
201 tensor_specs: Iterable[tf.TensorSpec], filepath_tensor: tf.Tensor218 for spec in tensor_specs
30 std::vector<OpSignatureTensorSpec> tensor_specs; in CreateOpSignatureTensorSpecs() local34 tensor_specs.push_back(tensor_spec); in CreateOpSignatureTensorSpecs()36 return tensor_specs; in CreateOpSignatureTensorSpecs()43 std::vector<OpSignatureTensorSpec> tensor_specs; in CreateOpSignatureTensorSpecs() local50 tensor_specs.push_back(tensor_spec); in CreateOpSignatureTensorSpecs()52 return tensor_specs; in CreateOpSignatureTensorSpecs()58 std::vector<OpSignatureTensorSpec> tensor_specs; in CreateOpSignatureTensorSpecs() local61 tensor_specs.push_back(tensor_spec); in CreateOpSignatureTensorSpecs()62 return tensor_specs; in CreateOpSignatureTensorSpecs()69 std::vector<OpSignatureTensorSpec> tensor_specs; in CreateOpSignatureTensorSpecs() local[all …]
45 std::vector<OpSignatureTensorSpec> tensor_specs; in GetOpSignatureTensorSpecs() local85 tensor_specs.push_back(tensor_spec); in GetOpSignatureTensorSpecs()87 return tensor_specs; in GetOpSignatureTensorSpecs()93 std::vector<OpSignatureTensorSpec> tensor_specs; in GetOpSignatureTensorSpecs() local117 tensor_specs.push_back(tensor_spec); in GetOpSignatureTensorSpecs()119 return tensor_specs; in GetOpSignatureTensorSpecs()
61 tensor_specs = get_node_tensor_specs(node)62 if tensor_specs is None:64 for tensor_spec in tensor_specs:
327 tensor_specs = []329 tensor_specs.append(tensor_spec.TensorSpec(s, dtypes.int32))331 def encode_tensor_specs(tensor_specs): argument332 trace_type.from_object(tensor_specs)336 lambda: encode_tensor_specs(tensor_specs), number=iterations)
548 tensor_specs = []552 tensor_specs.extend(spec._flat_tensor_specs) # pylint: disable=protected-access553 return tensor_specs
82 tensor_specs = [TensorSpec.from_dict(ts) for ts in header['features']]93 for ts in tensor_specs:
303 tensor_specs = checkpoint_utils.tff_type_to_tensor_spec_list(result_type)312 for var_name, spec in zip(var_names, tensor_specs)
1154 tensor_specs = []1160 tensor_specs.append(1163 *tensor_specs)