Searched refs:bool_tensor (Results 1 – 1 of 1) sorted by relevance
42 auto bool_tensor = in_tensors_.front(); in Run() local43 MS_ASSERT(bool_tensor != nullptr); in Run()44 MS_ASSERT(bool_tensor->data_type() == kNumberTypeBool); in Run()45 if (bool_tensor->Size() == 1) { in Run()46 auto condition = static_cast<bool *>(bool_tensor->data()); in Run()68 MS_ASSERT(bool_tensor->shape().size() == in_tensors_.at(1)->shape().size()); in Run()70 if (bool_tensor->shape()[i] != in_tensors_.at(1)->shape()[i]) { in Run()71 …ERROR) << "Tensor shapes differ in dim: " << i << " in_tensors_.at(0): " << bool_tensor->shape()[i] in Run()78 auto condition = static_cast<bool *>(bool_tensor->data()); in Run()