Searched defs:cond_shape (Results 1 – 9 of 9) sorted by relevance
37 const TensorShape cond_shape = ctx->InputShape(0); in Compile() local88 const TensorShape cond_shape = ctx->InputShape(0); in Compile() local
40 const RuntimeShape& cond_shape = GetTensorShape(cond_tensor); in ResizeOutputTensor() local
350 auto cond_shape = ShapeUtil::MakeTupleShape({s32_shape}); in TEST_F() local409 auto cond_shape = ShapeUtil::MakeTupleShape({s32_shape}); in TEST_F() local442 auto cond_shape = ShapeUtil::MakeTupleShape({s32_shape}); in TEST_F() local476 auto cond_shape = ShapeUtil::MakeTupleShape({s32_shape}); in TEST_F() local521 auto cond_shape = ShapeUtil::MakeTupleShape({s32_shape}); in TEST_F() local
76 const auto cond_shape = cond_shape_tensor.flat<int64_t>(); in Compute() local
782 Value cond_shape = GetShape(cond, op->getLoc(), rewriter); in rewriteOpWithDynamicInput() local840 auto cond_shape = cond.getType().cast<ShapedType>().getShape(); in matchAndRewrite() local
955 auto cond_shape = cond_tensor.getShape().front(); in verify() local
3004 ArrayRef<int64_t> cond_shape; in BuildSelectV2Op() local
2005 BHWC cond_shape, true_shape, false_shape; in Parse() local
2962 Value cond_shape = b.createOrFold<shape::ShapeOfOp>(op.condition()); in matchAndRewrite() local