Home
last modified time | relevance | path

Searched refs:input_output_alias_config (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dalias_passthrough_params_test.cc40 const auto& alias_config = module->input_output_alias_config(); in TEST_F()
56 const auto& alias_config = module->input_output_alias_config(); in TEST_F()
75 auto& preset_alias = module->input_output_alias_config(); in TEST_F()
81 const auto& alias_result = module->input_output_alias_config(); in TEST_F()
Dalias_passthrough_params.cc43 if (module->input_output_alias_config().OutputHasAlias({i}) || in Run()
44 module->input_output_alias_config().ParameterHasAlias( in Run()
51 TF_RETURN_IF_ERROR(module->input_output_alias_config().SetUpAlias( in Run()
Dgpu_executable.cc999 hlo_module.input_output_alias_config().GetAliasedParameter(index); in GetOutputInfo()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc75 hlo_module->input_output_alias_config(); in RunAndCheck()
257 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({0}, 0, {0})); in TEST_F()
258 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({1}, 0, {1})); in TEST_F()
274 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({0}, 0, {0})); in TEST_F()
275 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({1}, 0, {1})); in TEST_F()
332 TF_ASSERT_OK(module->get()->input_output_alias_config().SetUpAlias( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc276 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
278 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
282 ASSERT_IS_NOT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
319 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
321 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
325 ASSERT_IS_NOT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
407 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
409 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
Doptimize_input_output_buffer_alias.cc99 &module->input_output_alias_config(); in Run()
Dhlo_module.h380 HloInputOutputAliasConfig& input_output_alias_config() { in input_output_alias_config() function
383 const HloInputOutputAliasConfig& input_output_alias_config() const { in input_output_alias_config() function
Dcopy_insertion_test.cc1434 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1436 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1467 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1469 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1533 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1574 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1615 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
2549 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
2553 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
2576 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
[all …]
Dhlo_module.cc260 std::string serialized_aliasing = input_output_alias_config().ToShortString(); in ToCord()
318 *proto.mutable_input_output_alias() = input_output_alias_config().ToProto(); in ToProto()
878 module->input_output_alias_config() = input_output_alias_config(); in Clone()
Dcopy_insertion.cc57 !computation->parent()->input_output_alias_config().ParameterHasAlias( in IsReadonlyEntryParameterValue()
374 module->input_output_alias_config().ForEachAlias( in AddCopiesForAliasedInputOutputs()
425 TF_RETURN_IF_ERROR(module->input_output_alias_config().ForEachAliasWithStatus( in AddCopiesForAliasedInputOutputs()
1952 module->input_output_alias_config().OutputHasAlias(index) && in AddSpecialCaseCopies()
1955 module->input_output_alias_config().GetAliasedParameter(index); in AddSpecialCaseCopies()
Dhlo_live_range.cc231 !module.input_output_alias_config().ParameterHasAlias( in CalculateBufferStartEndMap()
Dhlo_live_range_test.cc240 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias({}, 0, {})); in TEST_F()
Dhlo_alias_analysis.cc55 module.input_output_alias_config(); in ComputeInputOutputAliasedValues()
Dhlo_runner.cc183 arguments, executable->module().input_output_alias_config(), in ExecuteWithDeviceBuffers()
Dhlo_computation.cc401 parent()->input_output_alias_config() = in set_root_instruction()
Dbuffer_assignment.cc1183 assignment->module().input_output_alias_config().ParameterHasAlias( in AssignSingleHloBuffer()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc73 executable->executable()->module().input_output_alias_config(); in Run()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate_registration.cc172 hlo_module->input_output_alias_config().ForEachAlias( in MlirHloToHloTextTranslateFunction()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_executable_interface.cc199 : hlo_module_->input_output_alias_config(); in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable_base.cc133 : hlo_module_->input_output_alias_config(); in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc309 executable->executable()->module().input_output_alias_config(), in RunExecutable()
373 executable->executable()->module().input_output_alias_config())); in RunExecutable()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc229 const HloInputOutputAliasConfig& config = module.input_output_alias_config(); in ComputeParametersThatMustBeDonated()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc347 executable->executable()->module().input_output_alias_config(); in ComputeAsync()
590 closure.executable()->executable()->module().input_output_alias_config(); in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc227 module().input_output_alias_config(); in CreateResultShapedBuffer()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc469 module->input_output_alias_config(), in TPUExecute()

12