Home
last modified time | relevance | path

Searched defs:sharding_key (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_compilation_cache_interface.cc369 std::vector<std::string>* proto_key, std::vector<std::string>* sharding_key, in CompileIfKeyAbsent()
407 std::vector<std::string>* proto_key, std::vector<std::string>* sharding_key, in CompileIfKeyAbsentHelper()
H A Dcompiled_subgraph.h128 std::vector<std::string> sharding_key; member
H A Dtpu_compile_op_common.cc286 std::vector<std::string> sharding_key; in ComputeInternal() local