Searched refs:proto_index (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_lookup.h | 60 virtual Status Lookup(int64 uid, int proto_index,
64 virtual Status Lookup(int64 uid, int proto_index, in Lookup() argument
66 return Lookup(uid, proto_index, std::move(entry), in Lookup()
|
| D | tpu_compilation_cache_local_lookup.cc | 49 int64 uid, int proto_index, in Lookup() argument
54 Status s = cache_->Lookup(uid, proto_index, entry); in Lookup()
55 VLOG(1) << "Looked up uid " << uid << ", index " << proto_index in Lookup()
|
| D | tpu_compilation_cache_rpc_lookup.cc | 99 int64 uid, int proto_index, in Lookup() argument
117 absl::StrCat(" _ ", uid, ":", proto_index, "_", in Lookup()
127 uid_and_index->set_proto_index(proto_index); in Lookup()
132 VLOG(1) << "Found uid " << uid << " and index " << proto_index in Lookup()
|
| D | tpu_compilation_cache_interface.cc | 557 int64 uid, int proto_index, in Lookup() argument
571 if (proto_index < 0 || in Lookup()
572 proto_index >= cache_entry->tpu_program_group->program_count()) { in Lookup()
573 return errors::NotFound("No proto found for core index ", proto_index, in Lookup()
577 proto_index); in Lookup()
595 int proto_index = iter->second.second; in Lookup() local
597 proto_index); in Lookup()
|
| D | tpu_compilation_cache_local_lookup.h | 39 Status Lookup(int64 uid, int proto_index,
|
| D | tpu_compilation_cache_rpc_lookup.h | 48 Status Lookup(int64 uid, int proto_index,
|
| D | tpu_compilation_cache_common.proto | 29 int32 proto_index = 2; field
|
| D | tpu_compilation_cache_service.cc | 101 call->request.uid_and_index().proto_index(), &entry); in GetTpuProgram()
|
| D | tpu_compilation_cache_interface.h | 146 Status Lookup(int64 uid, int proto_index,
|
| /external/perfetto/src/trace_processor/python/perfetto/trace_processor/ |
| D | api.py | 66 def get_cell_list(self, proto_index): argument
67 if proto_index == TraceProcessor.QUERY_CELL_NULL_FIELD_ID:
69 elif proto_index == TraceProcessor.QUERY_CELL_VARINT_FIELD_ID:
71 elif proto_index == TraceProcessor.QUERY_CELL_FLOAT64_FIELD_ID:
73 elif proto_index == TraceProcessor.QUERY_CELL_BLOB_FIELD_ID:
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | snapshot_util.cc | 249 int proto_index = 0; in WriteTensors() local
257 tensor_protos[proto_index].SerializeToArray( in WriteTensors()
259 proto_index++; in WriteTensors()
|