Home
last modified time | relevance | path

Searched defs:true_tensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dselect_v2_test_util.cc34 TrueTensor& true_tensor, ElseTensor& false_tensor, in SetUpData()
112 const TrueTensor& true_tensor, in RunSelectV2()
148 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2TestTemplate() local
176 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2BatchTestTemplate() local
206 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2BroadcastFalseTestTemplate() local
236 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2BroadcastTrueTestTemplate() local
266 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2BroadcastBothTestTemplate() local
294 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2ChannelsTestTemplate() local
323 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2ChannelsBatchTestTemplate() local
354 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2ChannelsBroadcastFalseTestTemplate() local
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtensor_indexing.cpp464 auto true_tensor = torch::tensor(1, torch::kUInt8); in TEST() local
488 auto true_tensor = torch::tensor(1, torch::kUInt8); in TEST() local
550 auto true_tensor = torch::tensor(true); in TEST() local
909 auto true_tensor = torch::tensor(true); in TEST() local
928 auto true_tensor = torch::tensor(true); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dsingle_machine_test.cc293 Tensor true_tensor(DT_BOOL, TensorShape()); in RunInfiniteTFLoop() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder.cc2000 const TfLiteTensor* true_tensor = reader->GetInputTensor(1); in Parse() local