Home
last modified time | relevance | path

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 Dselect_v2_test_util.cc34 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 Dtensor_indexing.cpp465 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 Dmodel_builder.cc2001 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 Dcontrol_flow_ops_test.py749 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 Doptimize_patterns.td846 // select(false_tensor, A, B) -> B