Searched refs:input_slices (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 145 input_slices = ops.IndexedSlices(
158 input_slices)
163 body_grad_graph, loop_vars, init_slices, input_slices, new_output_slices,
248 def _update_indexed_slices_param(graph, loop_vars, init_slices, input_slices, argument
284 graph.inputs = (graph.inputs[:flat_idx] + _flatten(input_slices) +
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 505 input_slices, argument
540 input_slices.values, communication_hint, timeout=timeout)
545 input_slices.indices, communication_hint, timeout=timeout)
549 dense_shape=input_slices.dense_shape)
553 densified = ops.convert_to_tensor(input_slices)
562 dense_shape=input_slices.dense_shape)
564 length = array_ops.shape(input_slices.indices)
|
| /external/perfetto/src/protozero/filtering/ |
| D | message_filter_unittest.cc | 772 std::vector<MessageFilter::InputSlice> input_slices; in TEST() local
777 input_slices.emplace_back( in TEST()
783 filt.FilterMessageFragments(input_slices.data(), input_slices.size()); in TEST()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional_test.py | 468 input_slices = array_ops.split(inputs, 4, axis=-1)
472 for inputs, weights in zip(input_slices, weight_slices)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distribute_strategy_test.py | 213 input_slices = (inputs, targets)
216 input_slices = inputs
219 original_dataset = (dataset_ops.Dataset.from_tensor_slices(input_slices))
|