Home
last modified time | relevance | path

Searched refs:dynamic_partition (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
Ddynamic_partition_op_test.py40 partitions = data_flow_ops.dynamic_partition(
63 partitions = data_flow_ops.dynamic_partition(
88 partitions = data_flow_ops.dynamic_partition(
110 partitions = data_flow_ops.dynamic_partition(
126 partitions = data_flow_ops.dynamic_partition(
139 partitions = data_flow_ops.dynamic_partition(
165 outputs = data_flow_ops.dynamic_partition(
185 partitions = data_flow_ops.dynamic_partition(
201 partitions = data_flow_ops.dynamic_partition(
217 partitions = data_flow_ops.dynamic_partition(
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_DynamicStitch.pbtxt46 This method can be used to merge partitions created by `dynamic_partition`
54 partitioned_data = tf.dynamic_partition(
57 condition_indices = tf.dynamic_partition(
Dapi_def_ParallelDynamicStitch.pbtxt45 This method can be used to merge partitions created by `dynamic_partition`
53 partitioned_data = tf.dynamic_partition(
56 condition_indices = tf.dynamic_partition(
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_grad.py36 partitioned_indices = data_flow_ops.dynamic_partition(
Dembedding_ops.py206 gather_ids = data_flow_ops.dynamic_partition(new_ids, p_assignments, np)
208 pindices = data_flow_ops.dynamic_partition(original_indices,
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dynamic_partition_op_test.py185 data_flow_ops.dynamic_partition(data_tensor, segment_ids_tensor,
Dragged_array_ops.py875 @dispatch.dispatch_for_api(data_flow_ops.dynamic_partition)
876 def dynamic_partition(data: ragged_tensor.RaggedOrDense, function
Dragged_dispatch_test.py995 op=data_flow_ops.dynamic_partition,
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py591 per_var_indices = data_flow_ops.dynamic_partition(local_indices,
601 per_var_values = data_flow_ops.dynamic_partition(indexed_slices.values,
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py589 done_indices, new_indices = data_flow_ops.dynamic_partition(
597 done_inp, new_inp = data_flow_ops.dynamic_partition(
4561 false_indices, true_indices = data_flow_ops.dynamic_partition(
4565 else_indices, then_indices = data_flow_ops.dynamic_partition(
4615 switch_indices = data_flow_ops.dynamic_partition(
4618 partitioned_indices = data_flow_ops.dynamic_partition(
4787 done_indices, new_indices = data_flow_ops.dynamic_partition(
4803 done_elem, new_elem = data_flow_ops.dynamic_partition(
4826 done_inp, new_inp = data_flow_ops.dynamic_partition(
/external/tensorflow/tensorflow/python/training/
Dinput.py713 data_flow_ops.dynamic_partition(x, select_i, num_partitions=2)[1]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt608 name: "dynamic_partition"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1188 name: "dynamic_partition"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td4050 This method can be used to merge partitions created by `dynamic_partition`
4058 partitioned_data = tf.dynamic_partition(
4061 condition_indices = tf.dynamic_partition(
10492 This method can be used to merge partitions created by `dynamic_partition`
10500 partitioned_data = tf.dynamic_partition(
10503 condition_indices = tf.dynamic_partition(