Searched refs:assert_op (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | remove_tensorflow_assert.cc | 31 const auto* assert_op = assert_it->get(); in Run() local
32 if (assert_op->type != OperatorType::kAssert) { in Run()
41 if (*it == assert_op->outputs[0]) { in Run()
49 CHECK(!CountOpsWithInput(*model, assert_op->outputs[0])); in Run()
55 LogName(*assert_op)); in Run()
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | test_util.py | 52 assert_op = check_ops.assert_equal(dx, expected_tensor, data=[dx])
53 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_parallelization_test.py | 42 assert_op = control_flow_ops.Assert(math_ops.greater(x, -1), [x])
43 with ops.control_dependencies([assert_op]):
|
| D | hoist_random_uniform_test.py | 50 assert_op = control_flow_ops.Assert(math_ops.greater_equal(y, 1), [y])
51 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/contrib/graph_editor/tests/ |
| D | transform_test.py | 79 assert_op = control_flow_ops.Assert(eq, [a, b])
80 with ops.control_dependencies([assert_op]):
82 sgv = ge.make_view([assert_op, eq.op, a.op, b.op])
85 new_assert_op = info.transformed(assert_op)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 1146 assert_op = _assert_rank_condition(x, rank, static_condition,
1157 return assert_op
1236 assert_op = _assert_rank_condition(x, rank, static_condition,
1247 return assert_op
1388 assert_op = _assert_ranks_condition(x, ranks, _static_rank_in,
1399 return assert_op
|
| D | image_ops_impl.py | 1630 assert_op = _assert(gamma >= 0, ValueError,
1632 if assert_op:
1633 gamma = control_flow_ops.with_dependencies(assert_op, gamma)
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | sampling_ops.py | 218 assert_op = control_flow_ops.Assert(
226 init_probs = control_flow_ops.with_dependencies([assert_op], init_probs)
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 259 assert_op = control_flow_ops.Assert(condition, [
263 with ops.control_dependencies([assert_op]):
321 assert_op = control_flow_ops.Assert(condition, [
325 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| D | receptive_field_test.py | 206 assert_op = control_flow_ops.Assert(
213 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/contrib/integrate/python/ops/ |
| D | odes.py | 203 assert_op = control_flow_ops.Assert(
206 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | state_space_model.py | 901 assert_op = control_flow_ops.Assert(
904 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 319 "assert_op.cc",
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 304 assert_op = tf.assert_greater_equal(input_tensor, input_tensor - 1)
305 with tf.control_dependencies([assert_op]):
|