Searched refs:AssignDeviceAndDataFormats (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer_test.cc | 402 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in ReduceTransposerKeepDims()
469 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in ReduceTransposerValidAxisNode()
515 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in TEST_F()
559 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in TEST_F()
599 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in TEST_F()
630 context.AssignDeviceAndDataFormats(kGPU, "ABCD", "ACBD"); in TEST_F()
668 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in TEST_F()
727 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in TEST_F()
781 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in TEST_F()
837 context.AssignDeviceAndDataFormats(kGPU, kSrcFormat, kDstFormat); in TEST_F()
[all …]
|
| H A D | generic_layout_optimizer.cc | 439 context.AssignDeviceAndDataFormats(kGPU, src_dst_formats.first, in Optimize()
446 context.AssignDeviceAndDataFormats(kCPU, kNCHW, kNHWC); in Optimize()
|
| H A D | generic_layout_optimizer_transposer.cc | 185 context_->AssignDeviceAndDataFormats(context_->target_device, in ScopedDataFormatUpgrader()
196 context_->AssignDeviceAndDataFormats(context_->target_device, in ~ScopedDataFormatUpgrader()
248 void TransposeContext::AssignDeviceAndDataFormats( in AssignDeviceAndDataFormats() function in tensorflow::grappler::TransposeContext
|
| H A D | generic_layout_optimizer_transposer.h | 65 void AssignDeviceAndDataFormats(absl::string_view target_device,
|