Searched refs:assert_op (Results 1 – 13 of 13) 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/data/experimental/kernel_tests/optimization/ |
| D | hoist_random_uniform_test.py | 48 assert_op = control_flow_ops.Assert(math_ops.greater_equal(y, 1), [y])
49 with ops.control_dependencies([assert_op]):
|
| D | map_parallelization_test.py | 39 assert_op = control_flow_ops.Assert(math_ops.greater(x, -1), [x])
40 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | test_util.py | 64 assert_op = check_ops.assert_equal(dx, expected_tensor)
65 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer_test.py | 101 assert_op = check_ops.assert_equal(dx, expected_tensor)
102 with ops.control_dependencies([assert_op]):
|
| D | loss_scale_test.py | 140 assert_op = check_ops.assert_equal(should_apply_gradients, is_finite)
143 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 1175 assert_op = _assert_rank_condition(x, rank, static_condition,
1186 return assert_op
1276 assert_op = _assert_rank_condition(x, rank, static_condition,
1287 return assert_op
1440 assert_op = _assert_ranks_condition(x, ranks, _static_rank_in,
1451 return assert_op
|
| D | image_ops_impl.py | 2219 assert_op = _assert(gamma >= 0, ValueError,
2221 if assert_op:
2222 gamma = control_flow_ops.with_dependencies(assert_op, gamma)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_rewrite_pass.cc | 561 auto assert_op = builder->create<TF::TPUCompileSucceededAssertOp>( in BuildTPUCompileSucceededAssertOp() local
563 WrapOpInLaunch(builder, compile_op->getLoc(), assert_op, compilation_device); in BuildTPUCompileSucceededAssertOp()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 65 assert_op = control_flow_ops.Assert(
69 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 762 assert_op = check_ops.assert_equal(x, in_tensor)
763 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | custom_training_loop_input_test.py | 345 assert_op = control_flow_ops.Assert(math_ops.less_equal(
347 with ops.control_dependencies([assert_op]):
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 354 "assert_op.cc",
|