Searched refs:false_tensor (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | select_v2_test_util.cc | 34 TrueTensor& true_tensor, ElseTensor& false_tensor, in SetUpData() argument
59 false_tensor.shape = in SetUpData()
67 false_tensor.data.push_back(false_value); in SetUpData()
86 false_tensor.data.push_back(expected_false_value); in SetUpData()
113 const ElseTensor& false_tensor, int batch, int height, in RunSelectV2() argument
131 else_descriptor.UploadData(false_tensor); in RunSelectV2()
149 Tensor<BHWC, DataType::FLOAT32> false_tensor; in SelectV2TestTemplate() local
151 cond_tensor, true_tensor, false_tensor, kBatch, kHeight, kWidth, in SelectV2TestTemplate()
161 false_tensor, kBatch, kHeight, kWidth, kChannels, in SelectV2TestTemplate()
177 Tensor<BHWC, DataType::FLOAT32> false_tensor; in SelectV2BatchTestTemplate() local
[all …]
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | tensor_indexing.cpp | 465 auto false_tensor = torch::tensor(0, torch::kUInt8); in TEST() local
480 assert_tensor_equal(torch::empty(sizes), a.index({false_tensor})); in TEST()
489 auto false_tensor = torch::tensor(0, torch::kUInt8); in TEST() local
504 a.index_put_({false_tensor}, 5); in TEST()
908 auto false_tensor = torch::tensor(false); in TEST() local
924 a.index({false_tensor, torch::tensor({0, 1}), "..."}), c10::Error); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | model_builder.cc | 2001 const TfLiteTensor* false_tensor = reader->GetInputTensor(2); in Parse() local
2004 false_tensor->allocation_type == kTfLiteMmapRo; in Parse()
2013 if (false_tensor->dims->size == 0) { in Parse()
2016 RETURN_IF_ERROR(ExtractTensorShape(*false_tensor, &false_shape)); in Parse()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | control_flow_ops_test.py | 749 fn_false, false_tensor = _build_false_branch(dtype)
758 false_tensor: np.ones([2, 2])
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | optimize_patterns.td | 846 // select(false_tensor, A, B) -> B
|