/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | alias_passthrough_params_test.cc | 40 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()
|
D | alias_passthrough_params.cc | 43 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()
|
D | gpu_executable.cc | 999 hlo_module.input_output_alias_config().GetAliasedParameter(index); in GetOutputInfo()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | buffer_donation_test.cc | 75 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/ |
D | hlo_alias_analysis_test.cc | 276 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()
|
D | optimize_input_output_buffer_alias.cc | 99 &module->input_output_alias_config(); in Run()
|
D | hlo_module.h | 380 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
|
D | copy_insertion_test.cc | 1434 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 …]
|
D | hlo_module.cc | 260 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()
|
D | copy_insertion.cc | 57 !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()
|
D | hlo_live_range.cc | 231 !module.input_output_alias_config().ParameterHasAlias( in CalculateBufferStartEndMap()
|
D | hlo_live_range_test.cc | 240 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias({}, 0, {})); in TEST_F()
|
D | hlo_alias_analysis.cc | 55 module.input_output_alias_config(); in ComputeInputOutputAliasedValues()
|
D | hlo_runner.cc | 183 arguments, executable->module().input_output_alias_config(), in ExecuteWithDeviceBuffers()
|
D | hlo_computation.cc | 401 parent()->input_output_alias_config() = in set_root_instruction()
|
D | buffer_assignment.cc | 1183 assignment->module().input_output_alias_config().ParameterHasAlias( in AssignSingleHloBuffer()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_compile_on_demand_op.cc | 73 executable->executable()->module().input_output_alias_config(); in Run()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | xla_mlir_translate_registration.cc | 172 hlo_module->input_output_alias_config().ForEachAlias( in MlirHloToHloTextTranslateFunction()
|
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
D | tpu_executable_interface.cc | 199 : hlo_module_->input_output_alias_config(); in ExecuteAsyncOnStream()
|
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
D | executable_base.cc | 133 : hlo_module_->input_output_alias_config(); in ExecuteAsyncOnStream()
|
/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
D | xrt_execute_op.cc | 309 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/ |
D | utils.cc | 229 const HloInputOutputAliasConfig& config = module.input_output_alias_config(); in ComputeParametersThatMustBeDonated()
|
/external/tensorflow/tensorflow/compiler/jit/kernels/ |
D | xla_ops.cc | 347 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/ |
D | cpu_executable.cc | 227 module().input_output_alias_config(); in CreateResultShapedBuffer()
|
/external/tensorflow/tensorflow/core/tpu/ |
D | tpu_execute.cc | 469 module->input_output_alias_config(), in TPUExecute()
|