Home
last modified time | relevance | path

Searched +full:absl +full:- +full:py (Results 1 – 25 of 609) sorted by relevance

12345678910>>...25

/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc7 http://www.apache.org/licenses/LICENSE-2.0
22 #include "absl/types/span.h"
41 namespace py = pybind11;
43 void BuildOpsSubmodule(py::module* m) { in BuildOpsSubmodule()
46 py::module ops = m->def_submodule("ops", "XLA operations"); in BuildOpsSubmodule()
48 py::enum_<TriangularSolveOptions::Transpose>( in BuildOpsSubmodule()
55 py::enum_<RandomAlgorithm>(ops, "RandomAlgorithm") in BuildOpsSubmodule()
60 py::enum_<CustomCallSchedule>(ops, "CustomCallSchedule") in BuildOpsSubmodule()
65 py::enum_<CustomCallApiVersion>(ops, "CustomCallApiVersion") in BuildOpsSubmodule()
72 ops.def("AfterAll", &AfterAll, py::arg("builder"), py::arg("tokens")); in BuildOpsSubmodule()
[all …]
Dpytree.cc7 http://www.apache.org/licenses/LICENSE-2.0
28 #include "absl/algorithm/container.h"
29 #include "absl/container/flat_hash_map.h"
30 #include "absl/hash/hash.h"
31 #include "absl/strings/str_cat.h"
32 #include "absl/strings/str_format.h"
33 #include "absl/strings/str_join.h"
43 namespace py = pybind11;
46 static auto* registry = []() -> PyTreeTypeRegistry* { in Singleton()
50 py::object type = py::reinterpret_borrow<py::object>( in Singleton()
[all …]
Dpmap_lib.cc7 http://www.apache.org/licenses/LICENSE-2.0
25 #include "absl/hash/hash.h"
26 #include "absl/strings/str_cat.h"
27 #include "absl/strings/str_format.h"
28 #include "absl/strings/str_join.h"
29 #include "absl/synchronization/notification.h"
30 #include "absl/types/span.h"
31 #include "absl/types/variant.h"
50 namespace py = pybind11;
58 InputSpec(ShardingSpec sharding_spec, py::object indices) in InputSpec()
[all …]
Dpython_ref_manager.cc7 http://www.apache.org/licenses/LICENSE-2.0
18 #include "absl/container/inlined_vector.h"
22 namespace py = pybind11;
25 PythonRefManager* manager, absl::Span<pybind11::object> objects) in ManagedPyObjects()
35 manager_->AddGarbage(absl::MakeSpan(objects_)); in ~ManagedPyObjects()
40 PythonRefManager::ManageReference(py::object object) { in ManageReference()
42 absl::Span<py::object>(&object, 1)); in ManageReference()
46 PythonRefManager::ManageReferences(absl::Span<py::object> objects) { in ManageReferences()
50 void PythonRefManager::AddGarbage(absl::Span<py::object> garbage) { in AddGarbage()
51 absl::MutexLock lock(&mu_); in AddGarbage()
[all …]
Dcallback.cc7 http://www.apache.org/licenses/LICENSE-2.0
29 namespace py = pybind11;
34 absl::Span<void* const> inputs(arg_ptrs, args_.size()); in PrepareAndCallInternal()
35 absl::Span<void* const> outputs(reinterpret_cast<void**>(result), in PrepareAndCallInternal()
38 py::gil_scoped_acquire gil; in PrepareAndCallInternal()
39 py::tuple args(inputs.size()); in PrepareAndCallInternal()
42 args[i] = py::none(); in PrepareAndCallInternal()
44 args[i] = py::array(args_[i].dtype, args_[i].dims, args_[i].strides, in PrepareAndCallInternal()
53 py::object output = py::reinterpret_borrow<py::object>( in PrepareAndCallInternal()
55 py::array array = py::cast<py::array>(std::move(output)); in PrepareAndCallInternal()
[all …]
Dpy_client.cc7 http://www.apache.org/licenses/LICENSE-2.0
22 #include "absl/base/casts.h"
23 #include "absl/container/flat_hash_map.h"
24 #include "absl/strings/numbers.h"
47 namespace py = pybind11;
55 buffers_.resize(pjrt_client_->device_count()); in PyClient()
56 for (PjRtDevice* device : pjrt_client_->addressable_devices()) { in PyClient()
57 if (device->id() >= buffers_.size()) { in PyClient()
58 buffers_.resize(device->id() + 1); in PyClient()
64 py::gil_scoped_release gil; in ~PyClient()
[all …]
Dxla.cc7 http://www.apache.org/licenses/LICENSE-2.0
22 #include "absl/strings/str_format.h"
23 #include "absl/strings/str_join.h"
24 #include "absl/types/span.h"
75 namespace py = pybind11;
91 py::register_exception<XlaRuntimeError>(m, "XlaRuntimeError", in PYBIND11_MODULE()
95 py::enum_<PrimitiveType>(m, "PrimitiveType") in PYBIND11_MODULE()
117 []() { return py::handle(tensorflow::Bfloat16Dtype()); }); in PYBIND11_MODULE()
122 py::class_<PjRtDevice, ClientAndPtr<PjRtDevice>>( in PYBIND11_MODULE()
130 "of this type, including remote devices on multi-host platforms.") in PYBIND11_MODULE()
[all …]
Djax_jit.cc7 http://www.apache.org/licenses/LICENSE-2.0
16 // This files implements the `jax.jit` dispatch and just-in-time feature.
20 // just-in-time compiled XLA Executable. All of that is done in C++ for
39 #include "absl/container/flat_hash_map.h"
40 #include "absl/container/inlined_vector.h"
41 #include "absl/strings/str_cat.h"
42 #include "absl/strings/str_format.h"
43 #include "absl/synchronization/notification.h"
44 #include "absl/types/span.h"
69 namespace py = pybind11;
[all …]
/external/tensorflow/tensorflow/python/framework/
Dop_def_library_pybind.cc7 http://www.apache.org/licenses/LICENSE-2.0
22 #include "absl/algorithm/container.h"
23 #include "absl/container/flat_hash_map.h"
24 #include "absl/container/flat_hash_set.h"
25 #include "absl/strings/match.h"
26 #include "absl/strings/str_cat.h"
27 #include "absl/strings/str_join.h"
40 namespace py = pybind11;
60 absl::flat_hash_map<std::string, absl::flat_hash_set<int>>;
62 using DefaultAttrMap = absl::flat_hash_map<std::string, py::object>;
[all …]
DBUILD14 # buildifier: disable=same-origin-load
17 # buildifier: disable=same-origin-load
20 # buildifier: disable=same-origin-load
23 # buildifier: disable=same-origin-load
26 # buildifier: disable=same-origin-load
36 "//third_party/py/tensorflow_numerics:__subpackages__",
49 copts = if_not_windows(["-Wno-sign-compare"]),
52 "-lm",
67 srcs = ["file_system_test.py"],
69 main = "file_system_test.py",
[all …]
/external/tink/testing/cross_language/
DBUILD.bazel12 srcs = ["key_generation_consistency_test.py"],
17 requirement("absl-py"),
34 srcs = ["key_version_test.py"],
39 requirement("absl-py"),
65 srcs = ["keyset_validation_test.py"],
70 requirement("absl-py"),
85 srcs = ["aead_test.py"],
90 requirement("absl-py"),
100 srcs = ["kms_aead_test.py"],
106 requirement("absl-py"),
[all …]
/external/tensorflow/tensorflow/tools/docs/
DBUILD6 # buildifier: disable=same-origin-load
38 srcs = ["tf_doctest_lib.py"],
45 "//third_party/py/numpy",
51 srcs = ["tf_doctest.py"],
52 args = ["--module_prefix_skip=" + ",".join(tpu_module + distribute_module + keras_module)],
68 "//third_party/py/numpy",
69 "@absl_py//absl/flags",
70 "@absl_py//absl/testing:absltest",
76 srcs = ["tf_doctest.py"],
77 args = ["--module=" + ",".join(tpu_module)],
[all …]
/external/tensorflow/tensorflow/python/util/
Dtf_stack.cc7 http://www.apache.org/licenses/LICENSE-2.0
24 // clang-format off
31 // clang-format on
39 #include "absl/algorithm/container.h"
40 #include "absl/container/flat_hash_set.h"
41 #include "absl/hash/hash.h"
42 #include "absl/strings/str_format.h"
43 #include "absl/strings/str_join.h"
44 #include "absl/types/span.h"
61 namespace py = pybind11;
[all …]
DBUILD5 # buildifier: disable=same-origin-load
8 # buildifier: disable=same-origin-load
11 # buildifier: disable=same-origin-load
14 # buildifier: disable=same-origin-load
26 "//third_party/py/cleverhans:__subpackages__",
27 "//third_party/py/launchpad:__subpackages__",
28 "//third_party/py/reverb:__subpackages__",
29 "//third_party/py/neural_structured_learning:__subpackages__",
30 "//third_party/py/tensorflow_examples:__subpackages__",
31 "//third_party/py/tf_agents:__subpackages__", # For benchmarks.
[all …]
/external/tensorflow/tensorflow/python/ops/ragged/
DBUILD11 #-------------------------------------------------------------------------------
13 #-------------------------------------------------------------------------------
17 srcs = ["__init__.py"],
52 srcs = ["ragged_array_ops.py"],
80 srcs = ["ragged_check_ops.py"],
90 srcs = ["ragged_batch_gather_ops.py"],
106 "ragged_batch_gather_with_default_op.py",
126 srcs = ["ragged_concat_ops.py"],
143 srcs = ["ragged_conversion_ops.py"],
153 srcs = ["ragged_factory_ops.py"],
[all …]
/external/tink/python/tink/jwt/
DBUILD.bazel10 srcs = ["__init__.py"],
32 srcs = ["_raw_jwt.py"],
42 srcs = ["_raw_jwt_test.py"],
46 requirement("absl-py"),
52 srcs = ["_verified_jwt.py"],
62 srcs = ["_verified_jwt_test.py"],
66 requirement("absl-py"),
72 srcs = ["_jwt_validator.py"],
82 srcs = ["_jwt_validator_test.py"],
87 requirement("absl-py"),
[all …]
/external/tink/python/tink/streaming_aead/
DBUILD.bazel10 srcs = ["__init__.py"],
22 srcs = ["_encrypting_stream.py"],
33 srcs = ["_encrypting_stream_test.py"],
38 requirement("absl-py"),
46 srcs = ["_decrypting_stream.py"],
57 srcs = ["_decrypting_stream_test.py"],
62 requirement("absl-py"),
69 srcs = ["_raw_streaming_aead.py"],
75 srcs = ["_streaming_aead.py"],
81 srcs = ["_streaming_aead_key_manager.py"],
[all …]
/external/tink/python/examples/walkthrough/
DBUILD.bazel10 srcs = ["create_keyset.py"],
19 srcs = ["create_keyset_test.py"],
22 requirement("absl-py"),
30 srcs = ["load_cleartext_keyset.py"],
39 srcs = ["load_cleartext_keyset_test.py"],
42 requirement("absl-py"),
51 srcs = ["load_encrypted_keyset.py"],
60 srcs = ["load_encrypted_keyset_test.py"],
63 requirement("absl-py"),
74 srcs = ["obtain_and_use_a_primitive.py"],
[all …]
/external/webrtc/modules/audio_processing/test/py_quality_assessment/
DBUILD.gn3 # Use of this source code is governed by a BSD-style license
24 "apm_quality_assessment.py",
26 "apm_quality_assessment_boxplot.py",
27 "apm_quality_assessment_export.py",
28 "apm_quality_assessment_gencfgs.py",
29 "apm_quality_assessment_optimize.py",
53 "quality_assessment/__init__.py",
54 "quality_assessment/annotations.py",
55 "quality_assessment/audioproc_wrapper.py",
56 "quality_assessment/collect_data.py",
[all …]
/external/tink/python/tink/
DBUILD.bazel10 srcs = ["__init__.py"],
31 srcs = ["_keyset_handle.py"],
46 srcs = ["_keyset_handle_test.py"],
51 requirement("absl-py"),
63 srcs = ["_keyset_reader.py"],
74 srcs = ["_keyset_reader_test.py"],
78 requirement("absl-py"),
86 srcs = ["_keyset_writer.py"],
97 srcs = ["_keyset_writer_test.py"],
101 requirement("absl-py"),
[all …]
/external/tensorflow/tensorflow/python/data/kernel_tests/
DBUILD4 load("//tensorflow:tensorflow.bzl", "cuda_py_test") # buildifier: disable=same-origin-load
5 load("//tensorflow:tensorflow.bzl", "tf_py_test") # buildifier: disable=same-origin-load
15 srcs = ["as_numpy_iterator_test.py"],
24 "//third_party/py/numpy",
25 "@absl_py//absl/testing:parameterized",
32 srcs = ["batch_test.py"],
46 "//third_party/py/numpy",
47 "@absl_py//absl/testing:parameterized",
54 srcs = ["bucket_by_sequence_length_test.py"],
64 "//third_party/py/numpy",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
DBUILD10 srcs = ["common.py"],
15 "@absl_py//absl:app",
16 "@absl_py//absl/flags",
17 "@absl_py//absl/logging",
23 srcs = ["common_v1.py"],
28 "@absl_py//absl:app",
29 "@absl_py//absl/flags",
30 "@absl_py//absl/logging",
38 "@llvm-project//llvm:FileCheck",
43 ["*.py"],
[all …]
/external/federated-compute/fcp/tensorflow/
DBUILD7 # http://www.apache.org/licenses/LICENSE-2.0
46 srcs = ["tf_py_smoke_test.py"],
54 # build is targeting a system-provided TF library or a bazel-built one.
101 "@com_google_absl//absl/base:core_headers",
102 "@com_google_absl//absl/container:flat_hash_map",
103 "@com_google_absl//absl/synchronization",
137 "@com_google_absl//absl/container:flat_hash_map",
138 "@com_google_absl//absl/strings",
139 "@com_google_absl//absl/strings:cord",
182 "@com_google_absl//absl/status",
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
DBUILD5 # buildifier: disable=same-origin-load
13 srcs = ["as_string_op_test.py"],
20 "//third_party/py/numpy",
27 srcs = ["base64_ops_test.py"],
36 "//third_party/py/numpy",
43 srcs = ["reduce_join_op_test.py"],
49 "//third_party/py/numpy",
56 srcs = ["regex_full_match_op_test.py"],
62 "@absl_py//absl/testing:parameterized",
69 srcs = ["regex_replace_op_test.py"],
[all …]
/external/tensorflow/tensorflow/core/tfrt/saved_model/tests/
DBUILD30 srcs = ["gen_resource_gather_v1.py"],
50 "@absl_py//absl:app",
51 "@absl_py//absl/flags",
57 srcs = ["gen_error_v1.py"],
71 "@absl_py//absl:app",
72 "@absl_py//absl/flags",
78 srcs = ["gen_pow.py"],
92 "@absl_py//absl:app",
93 "@absl_py//absl/flags",
99 srcs = ["gen_sparse_tensor_input.py"],
[all …]

12345678910>>...25