Searched refs:input_structure (Results 1 – 6 of 6) sorted by relevance
285 input_structure=structure.TensorStructure(dtypes.int64, []))302 input_structure=structure.NestedStructure(355 input_structure=self._state_structure)398 input_structure=structure.TensorStructure(dtypes.int64, []))420 input_structure = structure.NestedStructure(424 input_structure=input_structure)
258 f, "ChooseFastestV2", input_structure=nested_structure)
61 input_structure=structure.NestedStructure(
40 input_structure = dataset_ops.get_structure(input_dataset)41 self._structure = input_structure._batch(None) # pylint: disable=protected-access
68 input_structure = (like_inputs_t, initial_state)73 inputs_t, state0 = nest.pack_sequence_as(input_structure, flat_inputs)
1286 input_structure=structure_lib.NestedStructure(2239 input_structure=None, argument2274 if input_structure is None:2295 self._input_structure = input_structure2481 input_structure=self._init_structure)2486 input_structure=self._init_func.output_structure)2491 input_structure=self._init_func.output_structure)