Home
last modified time | relevance | path

Searched refs:guaranteed_constants (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_op_util.cc87 const OpInputList& guaranteed_constants) { in GuaranteedConstFingerprint() argument
90 for (const Tensor& constant : guaranteed_constants) { in GuaranteedConstFingerprint()
106 uint64 mlir_module_fingerprint, const OpInputList& guaranteed_constants, in CreateCompilationCacheKey() argument
133 guaranteed_constants.size(), in CreateCompilationCacheKey()
148 if (guaranteed_constants.size() > 0) { in CreateCompilationCacheKey()
156 key.guaranteed_const_fingerprint = [&metadata, &guaranteed_constants, in CreateCompilationCacheKey()
160 metadata.guaranteed_const_fingerprint(), guaranteed_constants); in CreateCompilationCacheKey()
Dtpu_compile_op_common.cc181 const GuaranteedConsts& guaranteed_constants, const XlaCompiler& compiler, in BuildComputationArgumentDescriptions() argument
208 guaranteed_constants.index() == 0 in BuildComputationArgumentDescriptions()
209 ? absl::get<0>(guaranteed_constants).size() in BuildComputationArgumentDescriptions()
210 : absl::get<1>(guaranteed_constants)->size(); in BuildComputationArgumentDescriptions()
214 if (guaranteed_constants.index() == 0) { in BuildComputationArgumentDescriptions()
219 *absl::get<0>(guaranteed_constants)[constant_count++])) in BuildComputationArgumentDescriptions()
225 (*absl::get<1>(guaranteed_constants))[constant_count++]; in BuildComputationArgumentDescriptions()
282 const GuaranteedConsts& guaranteed_constants, const NameAttrList& function, in CompileTFFunctionToHlo() argument
301 arg_shapes, guaranteed_constants, *compiler, &args, arg_core_mapping, in CompileTFFunctionToHlo()
595 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCache() argument
[all …]
Dtpu_compile_op_common.h123 const OpInputList& guaranteed_constants,
134 const OpInputList& guaranteed_constants, in LookupPersistentCompilationCacheAndFillCaches() argument
176 const GuaranteedConsts& guaranteed_constants,
200 const GuaranteedConsts& guaranteed_constants, const XlaCompiler& compiler,
Dtpu_compile_op_support.cc360 if (function_computation.guaranteed_constants.index() == 0) { in CreateTpuCompilationRequest()
361 absl::Span<const TensorProto* const> guaranteed_constants = in CreateTpuCompilationRequest() local
362 absl::get<0>(function_computation.guaranteed_constants); in CreateTpuCompilationRequest()
363 for (const TensorProto* constant : guaranteed_constants) { in CreateTpuCompilationRequest()
367 CHECK_EQ(function_computation.guaranteed_constants.index(), 1); in CreateTpuCompilationRequest()
368 const OpInputList& guaranteed_constants = in CreateTpuCompilationRequest() local
369 *absl::get<1>(function_computation.guaranteed_constants); in CreateTpuCompilationRequest()
370 for (const Tensor& constant : guaranteed_constants) { in CreateTpuCompilationRequest()
Dtpu_op_util.h37 uint64 mlir_module_fingerprint, const OpInputList& guaranteed_constants,
Dtpu_compile.proto48 repeated TensorProto guaranteed_constants = 7; field
Dtpu_compile_op_support.h62 GuaranteedConsts guaranteed_constants; member
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUCompile.pbtxt9 name: "guaranteed_constants"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUCompile.pbtxt11 'guaranteed_constants' is a list of tensors which have been guaranteed to not
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.h335 std::vector<Node*>* guaranteed_constants);
Ddistributed_tpu_rewrite_pass.cc2460 std::vector<Node*>* guaranteed_constants) { in FindGuaranteedConstantInputs() argument
2466 guaranteed_constants->push_back(input_edges[i]->src()); in FindGuaranteedConstantInputs()
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt53725 name: "guaranteed_constants"