| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_boolean_mask_op_test.py | 244 actual = ragged_array_ops.boolean_mask(data, mask)
251 ragged_array_ops.boolean_mask, [[1, 2]],
254 self.assertRaises(TypeError, ragged_array_ops.boolean_mask, [[1, 2]],
258 'RaggedTensor with dtype int32', ragged_array_ops.boolean_mask,
264 ragged_array_ops.boolean_mask, [[1, 2]],
269 ragged_array_ops.boolean_mask,
274 ragged_array_ops.boolean_mask, [[1, 2]], True)
277 ragged_array_ops.boolean_mask,
|
| D | ragged_conversion_ops.py | 89 values=array_ops.boolean_mask(grad, mask),
90 indices=array_ops.boolean_mask(indices, mask),
97 default_grads = array_ops.boolean_mask(grad, ~mask)
|
| D | ragged_array_ops.py | 45 def boolean_mask(data, mask, name=None): function
141 masked_values = boolean_mask(data, mask)
156 masked_lengths = array_ops.boolean_mask(lengths, mask)
165 masked_values = boolean_mask(data.values, segment_mask)
177 return boolean_mask(data, mask)
182 masked_values = array_ops.boolean_mask(data, mask)
|
| D | ragged_tensor.py | 1592 masked_data = array_ops.boolean_mask(tensor, dense_ragged_mask)
1679 values = array_ops.boolean_mask(tensor, mask)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resources.py | 104 return array_ops.boolean_mask(variable_names_tensor, variables_mask)
|
| D | ctc_ops.py | 1118 array_ops.boolean_mask(idx, flat_idx_mask), axis=1),
1119 updates=array_ops.boolean_mask(flat_labels, flat_label_mask),
1146 array_ops.boolean_mask(flat_indices, flat_mask), 1)
1147 values = array_ops.boolean_mask(flat_values, flat_mask)
|
| D | array_ops.py | 972 return boolean_mask(tensor=tensor, mask=slice_spec)
1773 def boolean_mask(tensor, mask, name="boolean_mask", axis=None): function
1922 return boolean_mask(tensor, mask, name, axis)
6473 masked = boolean_mask(tiled, mask)
|
| D | check_ops.py | 252 x_vals = array_ops.boolean_mask(x, mask)
253 y_vals = array_ops.boolean_mask(y, mask)
|
| D | variables.py | 3431 return array_ops.boolean_mask(variable_names_tensor, variables_mask)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.ragged.pbtxt | 4 name: "boolean_mask"
|
| D | tensorflow.pbtxt | 520 name: "boolean_mask"
521 …'mask\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'boolean_mask\'], "
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.ragged.pbtxt | 8 name: "boolean_mask"
|
| D | tensorflow.pbtxt | 940 name: "boolean_mask"
941 …'mask\', \'name\', \'axis\'], varargs=None, keywords=None, defaults=[\'boolean_mask\', \'None\'], "
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | quantized_ops_test.py | 74 gather_index = array_ops.boolean_mask(iota_r1, selected_index)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 146 masked_tensor = array_ops.boolean_mask(arr, mask, axis=axis)
203 tf_result = array_ops.boolean_mask(arr, mask)
213 tf_result = array_ops.boolean_mask(arr, mask)
240 array_ops.boolean_mask(ph_tensor, ph_mask),
255 array_ops.boolean_mask(tensor, mask)
262 array_ops.boolean_mask(tensor, mask).eval()
269 array_ops.boolean_mask(tensor, mask).eval()
276 array_ops.boolean_mask(tensor, mask).eval()
286 masked_tensor = array_ops.boolean_mask(string_tensor, bool_tensor)
298 return array_ops.boolean_mask([1, 2, 3], [True, False, True],
[all …]
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | local_test.py | 505 lc_2_kernel_flat = array_ops.boolean_mask(
524 lc_2_kernel_flat = array_ops.boolean_mask(
|
| D | core.py | 126 boolean_mask = K.any(
128 outputs = inputs * math_ops.cast(boolean_mask, inputs.dtype)
130 … outputs._keras_mask = array_ops.squeeze(boolean_mask, axis=-1) # pylint: disable=protected-access
1544 array_ops.boolean_mask,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 354 return array_ops.boolean_mask(tensor=tensor, mask=slice_spec)
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 414 return array_ops.boolean_mask(tensor=a_t, mask=condition_t, axis=axis)
1799 return array_ops.boolean_mask(tensor=self, mask=slice_spec)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 6294 label_ind = array_ops.boolean_mask(label_array, dense_mask)
6300 batch_ind = array_ops.boolean_mask(batch_array, dense_mask)
|
| /external/tensorflow/ |
| D | RELEASE.md | 2124 * Add `tf.ragged.boolean_mask`.
|