Searched refs:true_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | select_v2_test_util.cc | 34 TrueTensor& true_tensor, ElseTensor& false_tensor, in SetUpData() argument
57 true_tensor.shape = in SetUpData()
64 true_tensor.data.push_back(true_value); in SetUpData()
81 true_tensor.data.push_back(expected_true_value); in SetUpData()
112 const TrueTensor& true_tensor, in RunSelectV2() argument
130 true_descriptor.UploadData(true_tensor); in RunSelectV2()
148 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2TestTemplate() local
151 cond_tensor, true_tensor, false_tensor, kBatch, kHeight, kWidth, in SelectV2TestTemplate()
160 env, data_type, storage, precision, cond_tensor, true_tensor, in SelectV2TestTemplate()
176 Tensor<BHWC, DataType::FLOAT32> true_tensor; in SelectV2BatchTestTemplate() local
[all …]
|
| /external/pytorch/test/cpp/api/ |
| D | tensor_indexing.cpp | 464 auto true_tensor = torch::tensor(1, torch::kUInt8); in TEST() local
476 ASSERT_NE(a.data_ptr(), a.index({true_tensor}).data_ptr()); in TEST()
488 auto true_tensor = torch::tensor(1, torch::kUInt8); in TEST() local
502 a.index_put_({true_tensor}, neg_ones_expanded * 2); in TEST()
550 auto true_tensor = torch::tensor(true); in TEST() local
560 ASSERT_THROW(a.index_put_({true_tensor}, a_expanded), c10::Error); in TEST()
909 auto true_tensor = torch::tensor(true); in TEST() local
916 {true_tensor, in TEST()
918 true_tensor, in TEST()
919 true_tensor, in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 293 Tensor true_tensor(DT_BOOL, TensorShape()); in RunInfiniteTFLoop() local
294 true_tensor.flat<bool>()(0) = true; in RunInfiniteTFLoop()
295 true_tensor.AsProtoTensorContent( in RunInfiniteTFLoop()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder.cc | 2000 const TfLiteTensor* true_tensor = reader->GetInputTensor(1); in Parse() local
2002 const bool is_if_constant = true_tensor->allocation_type == kTfLiteMmapRo; in Parse()
2007 if (true_tensor->dims->size == 0) { in Parse()
2010 RETURN_IF_ERROR(ExtractTensorShape(*true_tensor, &true_shape)); in Parse()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops_test.py | 748 fn_true, true_tensor = _build_true_branch(dtype)
757 true_tensor: np.zeros([2, 2]),
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_patterns.td | 838 // select(true_tensor, A, B) -> A
|