Searched defs:condition_shape (Results 1 – 3 of 3) sorted by relevance
53 TensorShape condition_shape = shape; in select_condition_shape() local69 TensorShape condition_shape = detail::select_condition_shape(shape, has_same_same_rank); in setup() local88 …TensorType compute_target(const TensorShape &shape, const TensorShape &condition_shape, DataType d… in compute_target()127 …SimpleTensor<T> compute_reference(const TensorShape &shape, const TensorShape &condition_shape, Da… in compute_reference()
124 Shape condition_shape = ShapeUtil::MakeShape(xla::PRED, {}); in XLA_TEST_F() local
3216 TfLiteIntArray* condition_shape = in Validate() local