Home
last modified time | relevance | path

Searched refs:compiler_metadata (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compilation_cache_rpc_support.cc119 CompilerMetadataSerializedProto compiler_metadata; in SerializeCacheEntryToBufferSlices() local
120 auto cleanup_compiler_metadata = xla::MakeCleanup([&compiler_metadata]() { in SerializeCacheEntryToBufferSlices()
121 if (compiler_metadata.size > 0) { in SerializeCacheEntryToBufferSlices()
122 stream_executor::tpu::SerializedProto_Free(compiler_metadata); in SerializeCacheEntryToBufferSlices()
127 &compiler_metadata); in SerializeCacheEntryToBufferSlices()
132 compiler_metadata.bytes, compiler_metadata.size)) { in SerializeCacheEntryToBufferSlices()
Dtpu_compilation_cache.proto31 Blob compiler_metadata = 4; field
Dtpu_program_group.cc361 int index, CompilerMetadataSerializedProto* compiler_metadata) const { in SerializeCompilerMetadata()
366 tpu_programs_[index], compiler_metadata, status.c_status); in SerializeCompilerMetadata()
Dtpu_program_group.h151 int index, CompilerMetadataSerializedProto* compiler_metadata) const;
/external/tensorflow/tensorflow/core/tpu/
Dtpu_ops_c_api.h379 CompilerMetadataSerializedProto* compiler_metadata, TF_Status* status);