Home
last modified time | relevance | path

Searched refs:output_spec (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Deinsum_dense.py248 output_spec = split_string.group(3)
275 output_dim_map = {dim: (i + elided) for i, dim in enumerate(output_spec)}
278 output_dim_map = {dim: i for i, dim in enumerate(output_spec)}
292 for dim in output_spec:
296 "weight spec '%s.'" % (dim, output_spec, input_spec,
297 output_spec))
309 (dim, input_spec, output_spec))
315 for i, char in enumerate(output_spec)
319 if char not in output_spec:
322 (char, output_spec))
[all …]
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dcompression_ops.py39 def uncompress(element, output_spec): argument
51 flat_types = structure.get_flat_tensor_types(output_spec)
52 flat_shapes = structure.get_flat_tensor_shapes(output_spec)
55 return structure.from_tensor_list(output_spec, tensor_list)
Ddata_service_ops.py317 lambda x: compression_ops.uncompress(x, output_spec=element_spec),
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dhashing_test.py270 output_spec = layer.compute_output_signature(input_spec)
271 self.assertEqual(output_spec.shape.dims, input_shape.dims)
272 self.assertEqual(output_spec.dtype, dtypes.int64)
Dcategory_crossing_test.py245 output_spec = layer.compute_output_signature(input_specs)
246 self.assertEqual(output_spec.shape.dims[0], input_shapes[0].dims[0])
247 self.assertEqual(output_spec.dtype, dtypes.string)
/external/tensorflow/tensorflow/c/
Dops.cc50 const char* output_spec) { in TF_OpDefinitionBuilderAddOutput() argument
51 reinterpret_cast<OpDefBuilder*>(builder)->Output(output_spec); in TF_OpDefinitionBuilderAddOutput()
Dops.h187 TF_OpDefinitionBuilder* builder, const char* output_spec);