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_t uid, int proto_index,
64 virtual Status Lookup(int64_t 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_t 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_t 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 | 561 int64_t uid, int proto_index, in Lookup() argument
575 if (proto_index < 0 || in Lookup()
576 proto_index >= cache_entry->tpu_program_group->program_count()) { in Lookup()
577 return errors::NotFound("No proto found for core index ", proto_index, in Lookup()
581 proto_index); in Lookup()
599 int proto_index = iter->second.second; in Lookup() local
601 proto_index); in Lookup()
|
| D | tpu_compilation_cache_local_lookup.h | 39 Status Lookup(int64_t uid, int proto_index,
|
| D | tpu_compilation_cache_rpc_lookup.h | 48 Status Lookup(int64_t 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_t uid, int proto_index,
|
| /external/tensorflow/tensorflow/core/data/ |
| D | snapshot_utils.cc | 265 int proto_index = 0; in WriteTensors() local
273 tensor_protos[proto_index].SerializeToArray( in WriteTensors()
275 proto_index++; in WriteTensors()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 592 int proto_index = 0; in CreateFromProto() local
604 CHECK_GE(proto.dynamic_slice_sizes_size(), proto_index); in CreateFromProto()
605 slice_sizes[i][j] = proto.dynamic_slice_sizes(proto_index); in CreateFromProto()
606 proto_index += 1; in CreateFromProto()
624 CHECK_GE(proto.dynamic_slice_sizes_size(), proto_index); in CreateFromProto()
626 proto.dynamic_slice_sizes(proto_index); in CreateFromProto()
627 proto_index += 1; in CreateFromProto()
642 slice_sizes[i][j] = proto.dynamic_slice_sizes(proto_index); in CreateFromProto()
643 proto_index += 1; in CreateFromProto()
650 slice_sizes[0][j] = proto.dynamic_slice_sizes(proto_index); in CreateFromProto()
[all …]
|