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
|