| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| gradients.py | 46 flat_inputs = nest.flatten(inputs)
53 return gradient_ops.gradients(y, flat_inputs)
66 [output.dtype] * len(flat_inputs),
75 out.set_shape(output_tensor_shape.concatenate(flat_inputs[i].shape))
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| saving_utils.py | 75 flat_inputs = nest.flatten(inputs)
78 for input_tensor, input_name in zip(flat_inputs, flat_input_names):
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| xla.py | 312 flat_inputs = nest.flatten(inputs)
314 flat_inputs = [ops.convert_to_tensor(x) for x in flat_inputs]
324 flat_inputs = [
326 for i, x in enumerate(flat_inputs)
329 # Re-pack flat_inputs in same structure as 'inputs'.
331 structure=inputs, flat_sequence=flat_inputs)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| tpu.py | 756 flat_inputs = [
760 flat_inputs = [[ops.convert_to_tensor(x) for x in inp] for inp in flat_inputs]
763 flat_input_types = [x.dtype for x in flat_inputs[0]]
772 types = [x.dtype for x in flat_inputs[i]]
[all...] |
| /external/tensorflow/tensorflow/python/keras/engine/ |
| training_generator.py | 544 flat_inputs = nest.flatten(data)
545 if hasattr(flat_inputs[0], 'shape'):
546 return int(flat_inputs[0].shape[0]), False
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| functional_rnn.py | 71 def FlatCellStep(*flat_inputs):
73 inputs_t, state0 = nest.pack_sequence_as(input_structure, flat_inputs)
|
| /external/tensorflow/tensorflow/python/eager/ |
| function.py | [all...] |
| /external/mesa3d/src/intel/blorp/ |
| blorp_genX_exec.h | 637 sbe.ConstantInterpolationEnable = prog_data->flat_inputs;
670 sbe.ConstantInterpolationEnable = prog_data->flat_inputs;
690 sf.ConstantInterpolationEnable = prog_data->flat_inputs;
[all...] |
| /external/mesa3d/src/intel/compiler/ |
| brw_compiler.h | 721 uint32_t flat_inputs; member in struct:brw_wm_prog_data
[all...] |
| brw_fs.cpp | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| rnn.py | [all...] |
| control_flow_ops.py | 408 flat_inputs = [nest.flatten(v, expand_composites=True) for v in inputs]
410 for component in zip(*flat_inputs)]
[all...] |
| /external/mesa3d/src/intel/vulkan/ |
| genX_pipeline.c | 323 .ConstantInterpolationEnable = wm_prog_data->flat_inputs,
[all...] |
| /external/mesa3d/src/mesa/drivers/dri/i965/ |
| genX_state_upload.c | [all...] |
