| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_op_registry.h | 71 string compilation_device_name;
96 static void RegisterBackend(const string& compilation_device_name,
101 static std::vector<string> BackendNames();
104 static bool IsBackendRegistered(const string& name);
108 static void RegisterCompilationDevice(const string& device_name,
117 static bool GetCompilationDevice(const string& device_name,
128 const string& compilation_device_name,
133 static const std::unordered_set<string>* CompileTimeConstantInputs(
134 const string& op);
163 std::unordered_map<string, Backend> backends_ GUARDED_BY(mutex_)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| kernel_thunk.h | 20 #include <string>
50 const string& kernel_name, const HloInstruction* hlo_instruction);
55 const string& kernel_name() const { return kernel_name_; }
70 const string kernel_name_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| session.proto | 51 string name = 1;
84 string execution_platform = 5;
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| convert_computation.cc | 22 #include <string>
35 void RealMain(const string& mode, const string& path) {
43 string out;
|
| /external/tensorflow/tensorflow/contrib/tensorrt/log/ |
| trt_logger.h | 33 string name_;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_managed_allocator.h | 29 string Name() override { return "GpuManagedAllocator"; }
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_worker_cache.h | 29 const string& local_target);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| meta_optimizer.h | 35 string name() const override { return "meta_optimizer"; };
46 std::unique_ptr<GraphOptimizer> NewOptimizer(const string& optimizer);
49 std::vector<std::pair<string, string>> result_;
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| grappler_test.h | 31 const std::vector<string>& node_names);
|
| /external/tensorflow/tensorflow/core/kernels/data/sql/ |
| driver_manager.cc | 23 const string& driver_name) {
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| string_split_fuzz.cc | 39 // of handing it a > int32_max size string, which should be tested for in
46 delimeter_tensor.scalar<string>()() =
47 string(reinterpret_cast<const char*>(data), delim_len);
48 input_tensor.scalar<string>()() = string(
|
| /external/tensorflow/tensorflow/core/kernels/ |
| i_remote_fused_graph_ops_definitions.h | 36 // Return op id for given string op name
37 virtual int GetOpIdFor(const string& op_type,
|
| lookup_table_init_op.h | 25 Status InitializeTableFromTextFile(const string& filename, int64 vocab_size,
|
| /external/tensorflow/tensorflow/core/platform/default/ |
| strong_hash.h | 24 inline uint64 StrongKeyedHash(const uint64 (&key)[2], const string& s) {
|
| /external/tensorflow/tensorflow/core/platform/posix/ |
| load_library.cc | 43 string FormatLibraryFileName(const string& name, const string& version) {
44 string filename;
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| tfprof_show_test.cc | 35 string CheckAndRemoveDoc(const string& doc) {
44 string graph_path =
53 string run_meta_path =
60 string op_log_path =
65 string ckpt_path = io::JoinPath(testing::TensorFlowSrcRoot(),
78 string TestToFromProto(const string& cmd, const Options& opts,
80 string profile_file = io::JoinPath(testing::TmpDir(), "profile");
89 string dump_str
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| equal_graph_def.h | 40 string* diff, const EqualGraphDefOptions& options = {});
57 bool EqualNodeDef(const NodeDef& actual, const NodeDef& expected, string* diff,
75 string* diff,
91 string diff; \
|
| events_writer_test.cc | 39 const string& tag, float simple_value) {
56 string record;
64 void VerifyFile(const string& filename) {
107 string GetDirName(const string& suffix) {
112 string file_prefix = GetDirName("/writeflush_test");
116 string filename = writer.FileName();
121 string file_prefix = GetDirName("/writeclose_test");
125 string filename = writer.FileName();
130 string file_prefix = GetDirName("/writedelete_test")
[all...] |
| stat_summarizer.h | 25 #include <string>
162 // Returns a string detailing the accumulated runtime stats in a tab-separated
164 std::string GetOutputString() const;
166 std::string ShortSummary() const;
168 // Prints the string returned by GetOutputString().
174 void ComputeStatsByType(std::map<string, int64>* node_type_map_count,
175 std::map<string, int64>* node_type_map_time,
176 std::map<string, int64>* node_type_map_memory,
177 std::map<string, int64>* node_type_map_times_called,
180 std::string GetStatsByNodeType() const
[all...] |
| /external/tensorflow/tensorflow/examples/android/jni/object_tracking/ |
| object_model.h | 53 ObjectModelBase(const std::string& name) : name_(name) {}
76 inline const std::string& GetName() const {
81 const std::string name_;
91 const std::string& name)
|
| /external/tensorflow/tensorflow/java/src/gen/cc/ |
| source_writer.cc | 16 #include <string>
33 SourceWriter& SourceWriter::Write(const string& str) {
38 if (line_pos != string::npos) {
45 } while (line_pos != string::npos && line_pos < str.size());
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| benchmark_model.h | 27 string name;
34 Status InitializeSession(int num_threads, const string& graph,
40 const std::vector<string>& outputs, Session* session,
46 const std::vector<string>& outputs, Session* session,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| freeze_requantization_ranges.cc | 25 string name;
32 Status ExtractMinMaxRecords(const string& log_file_name,
34 string file_data;
37 const string print_suffix("__print__");
38 const string requant_prefix("__requant_min_max:");
39 std::vector<string> file_lines = str_util::Split(file_data, '\n');
40 for (const string& file_line : file_lines) {
47 std::vector<string> line_parts = str_util::Split(file_line, ';');
67 string min_max_string = line_parts[min_max_index];
68 std::vector<string> min_max_parts = str_util::Split(min_max_string, '[')
[all...] |
| transform_utils.cc | 32 std::set<string>* matched_nodes) {
39 inline uint64 Hash64String(const string& input) {
45 std::set<string> found_nodes;
64 std::map<string, const NodeDef*>* result) {
71 std::map<string, std::vector<const NodeDef*>>* result) {
72 std::map<string, const NodeDef*> node_map;
75 for (const string& input : node.input()) {
76 string input_node_name = NodeNameFromInput(input);
82 void NodeNamePartsFromInput(const string& input_name, string* prefix
[all...] |
| /external/v8/src/arm/ |
| codegen-arm.h | 16 // Generates the code for handling different string types and loading the
20 Register string,
|
