Home
last modified time | relevance | path

Searched refs:test_ops (Results 1 – 25 of 51) sorted by relevance

123

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dplacement_test.py27 from tensorflow.python.framework import test_ops
213 dataset_device = test_ops.device_placement_op()
217 iterator_device = test_ops.device_placement_op()
221 data_device = test_ops.device_placement_op()
225 get_value_device = test_ops.device_placement_op()
229 has_value_device = test_ops.device_placement_op()
242 dataset_device = test_ops.device_placement_op()
246 iterator_device = test_ops.device_placement_op()
250 data_device = test_ops.device_placement_op()
254 get_value_device = test_ops.device_placement_op()
[all …]
Dscan_test.py30 from tensorflow.python.framework import test_ops
279 device = test_ops.device_placement_op()
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dtfr_gen_test.py24 from tensorflow.compiler.mlir.tfr.resources import gen_test_ops as test_ops unknown
160 return test_ops.TestTwoInputsOp(ins[0], ins[1])
266 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_loc', [test_ops])
303 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_tensor', [test_ops])
381 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_control_flow', [test_ops])
442 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_tf_ops', [test_ops])
533 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_attrs', [test_ops])
574 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_shapes', [test_ops])
607 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_temp', [test_ops])
618 mlir_code = tfr_gen(sys.modules[__name__], '_tfr_quant', [test_ops])
/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
Dcontrol_flow_util_test.py20 from tensorflow.python.framework import test_ops
38 ref_switch_false, _ = control_flow_ops.ref_switch(test_ops.ref_output(),
43 self.assertFalse(control_flow_util.IsSwitch(test_ops.int_output().op))
51 ref_enter = gen_control_flow_ops.ref_enter(test_ops.ref_output(),
61 self.assertFalse(control_flow_util.IsLoopEnter(test_ops.int_output().op))
68 ref_exit = control_flow_ops.exit(test_ops.ref_output()).op
71 self.assertFalse(control_flow_util.IsLoopExit(test_ops.int_output().op))
Dcond_v2_test.py30 from tensorflow.python.framework import test_ops
1560 cpu_zero_op = test_ops.device_placement_op()
1563 cpu_one_op = test_ops.device_placement_op()
1581 return test_ops.device_placement_op()
1608 local_op = test_ops.device_placement_op()
1610 worker_op = test_ops.device_placement_op()
1627 local_op = test_ops.device_placement_op()
1629 worker_op = test_ops.device_placement_op()
1644 result = test_ops.device_placement_op()
1652 op_on_cpu_0 = test_ops.device_placement_op(name="colocation_test_op")
[all …]
DBUILD26 "//tensorflow/python:test_ops",
86 "//tensorflow/python:test_ops",
/external/grpc-grpc/test/core/end2end/tests/
Dcancel_after_invoke.cc39 size_t test_ops, in begin_test() argument
44 config.name, mode.name, test_ops); in begin_test()
95 cancellation_mode mode, size_t test_ops) { in test_cancel_after_invoke() argument
100 mode, test_ops, nullptr, nullptr); in test_cancel_after_invoke()
159 error = grpc_call_start_batch(c, ops, test_ops, tag(1), nullptr); in test_cancel_after_invoke()
Dcancel_before_invoke.cc91 size_t test_ops) { in test_cancel_before_invoke() argument
96 begin_test(config, "cancel_before_invoke", test_ops, nullptr, nullptr); in test_cancel_before_invoke()
157 error = grpc_call_start_batch(c, ops, test_ops, tag(1), nullptr); in test_cancel_before_invoke()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dispatch_test.py41 from tensorflow.python.ops.ragged import ragged_tensor_test_ops as test_ops unknown
93 for op in test_ops.UNARY_FLOAT_OPS] +
96 for op in test_ops.UNARY_BOOL_OPS] +
99 for op in test_ops.UNARY_INT_OPS] +
103 for op in test_ops.UNARY_STRING_OPS] +
332 for op in test_ops.BINARY_FLOAT_OPS] +
336 for op in test_ops.BINARY_INT_OPS] +
340 for op in test_ops.BINARY_BOOL_OPS] +
471 for op in test_ops.BINARY_ASSERT_OPS] +
475 for op in test_ops.BINARY_ASSERT_OPS] +
[all …]
Dragged_tensor_supported_values_test.py36 from tensorflow.python.ops.ragged import ragged_tensor_test_ops as test_ops unknown
193 for op in test_ops.UNARY_FLOAT_OPS] +
196 for op in test_ops.UNARY_BOOL_OPS] +
200 for op in test_ops.UNARY_INT_OPS] +
203 for op in test_ops.UNARY_STRING_OPS] +
274 for op in test_ops.BINARY_FLOAT_OPS] +
278 for op in test_ops.BINARY_INT_OPS] +
282 for op in test_ops.BINARY_BOOL_OPS]
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/customization/
Dtest_ops_test.py20 from tensorflow.python.framework import test_ops
27 attr = tf.function(test_ops.test_attr)(T=tf.float32)
DBUILD19 name = "test_ops",
41 "//tensorflow/python:test_ops",
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_test.py32 from tensorflow.python.framework import test_ops
306 fastpath_dtype = test_ops.dtype_with_default_op(im).numpy()
307 slowpath_dtype = test_ops.dtype_with_default_op_eager_fallback(
313 graph_dtype_symbolic = test_ops.dtype_with_default_op(im)
324 return test_ops.dtype_with_default_op(im)
332 return test_ops.dtype_with_default_op(im)
/external/tensorflow/tensorflow/python/framework/
Dops_test.py50 from tensorflow.python.framework import test_ops
72 pt = test_ops.stub_resource_handle_op(container="a", shared_name="b")
73 test_ops.resource_create_op(pt).run()
78 handle = test_ops.stub_resource_handle_op(container="a", shared_name="b")
81 create_op=test_ops.resource_create_op(handle),
82 is_initialized_op=test_ops.resource_initialized_op(handle))
675 op = test_ops.float_output_string_output(name="myop").a.op
698 op1 = test_ops.float_output(name="myop1").op
700 op2 = test_ops.float_input(float_t, name="myop2")
714 op1 = test_ops.float_output(name="myop1").op
[all …]
Dfunction_def_to_graph_test.py27 from tensorflow.python.framework import test_ops
149 list_output0, list_output1 = test_ops.list_output(
152 d_2, e_2 = test_ops.foo1(a=d_1, b=e_1, c=list_output1, name="foo_2")
DBUILD420 ":test_ops",
1078 tags = ["no_pip"], # test_ops are not available in pip.
1087 ":test_ops",
1107 tags = ["no_pip"], # test_ops are not available in pip.
1111 ":test_ops",
1636 ":test_ops",
1696 ":test_ops",
1711 ":test_ops",
1719 name = "test_ops",
1720 out = "test_ops.py",
[all …]
Dconfig_test.py30 from tensorflow.python.framework import test_ops
163 return test_ops.test_attr(T=dtypes.float32, name='test_attr')
340 op = test_ops.device_placement_op()
879 self.assertTrue(test_ops.is_tensor_float32_enabled())
882 self.assertFalse(test_ops.is_tensor_float32_enabled())
885 self.assertTrue(test_ops.is_tensor_float32_enabled())
/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
Dgen_saved_model_v1.py22 from tensorflow.python.framework import test_ops
62 sleep_op = test_ops.sleep_identity_op(1, x1, name='sleep')
/external/tensorflow/tensorflow/python/ops/
Daccumulate_n_benchmark.py99 test_ops = (("AddN", add_n.op),
105 for tag, op in test_ops:
/external/skia/tests/
DPoint3Test.cpp27 static void test_ops(skiatest::Reporter* reporter) { in test_ops() function
128 test_ops(reporter); in DEF_TEST()
/external/skia/tools/gpu/
DTestOps.h18 namespace sk_gpu_test::test_ops {
/external/skia/gm/
Dtexelsubset.cpp170 if (auto op = sk_gpu_test::test_ops::MakeRect(rContext, in onDraw()
191 if (auto op = sk_gpu_test::test_ops::MakeRect(rContext, std::move(fp2), in onDraw()
Daarecteffect.cpp88 auto op = sk_gpu_test::test_ops::MakeRect(rContext, std::move(grPaint), drawRect); in onDraw()
Dconvexpolyeffect.cpp123 auto op = sk_gpu_test::test_ops::MakeRect(rContext, std::move(grPaint), rect); in onDraw()
/external/igt-gpu-tools/tools/
Dskl_compute_wrpll.c856 struct test_ops { struct
863 static void test_run(struct test_ops *test) in test_run() argument

123