| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_boolean_mask_op_test.py | 313 actual = ragged_array_ops.boolean_mask(data, mask, keepdims=keepdims)
320 ragged_array_ops.boolean_mask, [[1, 2]],
323 self.assertRaises(TypeError, ragged_array_ops.boolean_mask, [[1, 2]],
327 'RaggedTensor with dtype int32', ragged_array_ops.boolean_mask,
333 ragged_array_ops.boolean_mask, [[1, 2]], [[True, False, True]])
337 ragged_array_ops.boolean_mask,
342 ragged_array_ops.boolean_mask, [[1, 2]], True)
345 ragged_array_ops.boolean_mask,
|
| D | ragged_array_ops.py | 37 def boolean_mask(data, mask, keepdims=False, name=None): function
162 masked_values = boolean_mask(data, mask, keepdims)
178 masked_lengths = array_ops.boolean_mask(lengths, mask)
187 masked_values = boolean_mask(data.values, segment_mask, keepdims=False)
197 return boolean_mask(data, mask, keepdims)
202 masked_values = array_ops.boolean_mask(data, mask)
|
| D | ragged_util.py | 188 masked = array_ops.boolean_mask(tiled, mask)
|
| D | ragged_tensor.py | 1194 masked_data = array_ops.boolean_mask(tensor, dense_ragged_mask)
1209 values = array_ops.boolean_mask(tensor, mask)
|
| /external/tensorflow/tensorflow/contrib/text/python/ops/ |
| D | skip_gram_ops.py | 416 input_tensor = array_ops.boolean_mask(input_tensor, mask)
417 freq = array_ops.boolean_mask(freq, mask)
449 return array_ops.boolean_mask(input_tensor, mask)
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_optimizer.py | 214 flat_ids = array_ops.boolean_mask(flat_ids, is_id_valid)
215 example_ids = array_ops.boolean_mask(example_ids, is_id_valid)
216 weight_tensor = array_ops.boolean_mask(weight_tensor, is_id_valid)
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/ |
| D | __init__.py | 121 boolean_mask = _ops.boolean_mask variable
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | rnn_common.py | 243 activations_masked = array_ops.boolean_mask(activations, mask)
244 labels_masked = array_ops.boolean_mask(labels, mask)
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| D | histogram_ops.py | 134 array_ops.boolean_mask(scores, boolean_labels),
141 array_ops.boolean_mask(scores, math_ops.logical_not(boolean_labels)),
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | reshape.py | 254 x_event_shape_specified = array_ops.boolean_mask(x_event_shape,
256 event_shape_in_specified = array_ops.boolean_mask(event_shape_in,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resources.py | 104 return array_ops.boolean_mask(variable_names_tensor, variables_mask)
|
| D | ctc_ops.py | 838 array_ops.boolean_mask(idx, flat_idx_mask), axis=1),
839 updates=array_ops.boolean_mask(flat_labels, flat_label_mask),
867 array_ops.boolean_mask(flat_indices, flat_mask), 1)
868 values = array_ops.boolean_mask(flat_values, flat_mask)
|
| D | check_ops.py | 482 x_vals = array_ops.boolean_mask(x, mask)
483 y_vals = array_ops.boolean_mask(y, mask)
|
| D | array_ops.py | 589 return boolean_mask(tensor=tensor, mask=slice_spec)
1275 def boolean_mask(tensor, mask, name="boolean_mask", axis=None): function
1400 return boolean_mask(tensor, mask, name, axis)
|
| D | variables.py | 3007 return array_ops.boolean_mask(variable_names_tensor, variables_mask)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/ |
| D | graph_io.py | 322 keys = array_ops.boolean_mask(keys, mask)
323 examples_proto = array_ops.boolean_mask(examples_proto, mask)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | quantized_ops_test.py | 73 gather_index = array_ops.boolean_mask(iota_r1, selected_index)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 136 masked_tensor = array_ops.boolean_mask(arr, mask, axis=axis)
187 tf_result = array_ops.boolean_mask(arr, mask)
196 tf_result = array_ops.boolean_mask(arr, mask)
222 array_ops.boolean_mask(ph_tensor, ph_mask),
237 array_ops.boolean_mask(tensor, mask)
244 array_ops.boolean_mask(tensor, mask).eval()
251 array_ops.boolean_mask(tensor, mask).eval()
258 array_ops.boolean_mask(tensor, mask).eval()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 93 boolean_mask = K.any(
95 outputs = inputs * math_ops.cast(boolean_mask, inputs.dtype)
97 … outputs._keras_mask = array_ops.squeeze(boolean_mask, axis=-1) # pylint: disable=protected-access
|
| D | local_test.py | 437 lc_2_kernel_flat = keras.backend.array_ops.boolean_mask(
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops_test.py | 930 masked_lt = ops.boolean_mask(self.original_lt, mask)
935 masked_lt = ops.boolean_mask(self.original_lt, mask)
937 array_ops.boolean_mask(self.original_lt.tensor, mask.tensor),
944 ops.boolean_mask(self.original_lt, mask)
949 ops.boolean_mask(self.original_lt, mask)
|
| D | ops.py | 1215 def boolean_mask(labeled_tensor, mask, name=None): function
1245 op = array_ops.boolean_mask(labeled_tensor.tensor, mask.tensor, name=scope)
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | hierarchical_controller.py | 356 selected_reward = array_ops.boolean_mask(ctr["reward"]["value"],
383 selected_loss = array_ops.boolean_mask(ctr["loss"], mask)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 468 name: "boolean_mask"
469 …'mask\', \'axis\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'boolean_mask\'], "
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5123 label_ind = array_ops.boolean_mask(label_array, dense_mask)
5129 batch_ind = array_ops.boolean_mask(batch_array, dense_mask)
|