Searched refs:input_slices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 136 input_slices = ops.IndexedSlices(
149 input_slices)
154 body_grad_graph, loop_vars, init_slices, input_slices, new_output_slices,
239 def _update_indexed_slices_param(graph, loop_vars, init_slices, input_slices, argument
272 graph.inputs = (graph.inputs[:flat_idx] + _flatten(input_slices) +
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | keras_test.py | 234 input_slices = (inputs, targets)
237 input_slices = inputs
241 input_slices))
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_estimator.py | 1007 input_slices = [
1013 broadcasted_inputs.append([x[core_id] for x in input_slices])
|