Home
last modified time | relevance | path

Searched defs:compilation_key (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_extract_outside_compilation.cc156 ValueRange inputs, Value compilation_key, in CreateSendFromHostOp()
179 TypeRange output_types, Value compilation_key, in CreateRecvAtHostOp()
508 Operation* insertion_point, Value compilation_key, in MoveOpsToHost()
591 Operation* insertion_point, Value compilation_key, in MoveOpsToHost()
647 Value compilation_key, Value device_ordinal, in DecomposeControlFlow()
740 Value compilation_key = compilation_key_op.program(); in CreateParallelExecuteForOutsideCompilation() local
H A Dtpu_dynamic_layout_pass.cc125 Value compilation_key, in BuildGetLayout()
166 const int64_t execute_arg_index, Value compilation_key, in HandleReplicatedInputs()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dmove_compilation_to_host.cc93 mlir::Value compilation_key = *compile_op.program().begin(); in CreateSendRecvOpsToTransferProgramKey() local
226 for (auto compilation_key : compilation_key_ops) { in HandleCompilationOps() local
H A Ddtensor_send_recv.cc41 [&](mlir::TF::_TPUCompileMlirPlaceholderProgramKeyOp compilation_key) { in GetOrCreateCompilationKey()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_ops.cc543 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() local
561 Tensor compilation_key(cpu_allocator, DT_STRING, TensorShape({})); in Compute() local