Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
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
/external/pytorch/test/cpp/api/
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
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc293 Tensor true_tensor(DT_BOOL, TensorShape()); in RunInfiniteTFLoop() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc2000 const TfLiteTensor* true_tensor = reader->GetInputTensor(1); in Parse() local