Searched defs:guaranteed_constants (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_op_util.cc | 88 const OpInputList& guaranteed_constants) { in GuaranteedConstFingerprint()
107 uint64 mlir_module_fingerprint, const OpInputList& guaranteed_constants, in CreateCompilationCacheKey()
|
| D | tpu_compile_op_common.cc | 167 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCache()
182 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCacheInternal()
236 OpInputList guaranteed_constants; in ComputeInternal() local
|
| D | tpu_compile.proto | 48 repeated TensorProto guaranteed_constants = 7; field
|
| D | tpu_compile_op_support.h | 64 GuaranteedConsts guaranteed_constants; member
|
| D | tpu_compile_op_common.h | 142 const OpInputList& guaranteed_constants, in LookupPersistentCompilationCacheAndFillCaches()
|
| D | tpu_compile_op_support.cc | 367 absl::Span<const TensorProto* const> guaranteed_constants = in CreateTpuCompilationRequest() local
374 const OpInputList& guaranteed_constants = in CreateTpuCompilationRequest() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.cc | 254 const GuaranteedConsts& guaranteed_constants, const XlaCompiler& compiler, in BuildComputationArgumentDescriptions()
375 const GuaranteedConsts& guaranteed_constants, const NameAttrList& function, in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2751 std::vector<Node*>* guaranteed_constants) { in FindGuaranteedConstantInputs()
|