Searched defs:shape_op (Results 1 – 14 of 14) sorted by relevance
77 Block* replicate_block, TF::ShapeOp shape_op) { in MakeShapeOpInvariant()158 replicate_op.walk([&](TF::ShapeOp shape_op) { in HoistReplicateInvariantOps()
380 auto shape_op = builder->create<TF::ShapeOp>( in BuildCompileOp() local
537 auto shape_op = shape_value.getDefiningOp(); in GetConstShapeValue() local
63 AbstractOperationPtr shape_op(ctx->CreateOperation()); in Shape() local
81 auto* shape_op = static_cast<TensorFlowShapeOperator*>(op); in Run() local
995 shape_op = lambda x: array_ops.shape_internal(x, optimize=False) function
684 auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); in ConvertOp() local745 auto shape_op = builder.create<tfl::ConstOp>(loc, output_shape); in ConvertOp() local
455 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST() local
573 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST() local
395 Output shape_op = in TestFoldFusedBatchNormsWithConcat() local
373 Output shape_op = in TestQuantizeConcat() local
238 auto shape_op = dyn_cast_or_null<ShapeOp>(slice_op.input().getDefiningOp()); in fold() local1908 auto shape_op = input().getDefiningOp<ShapeOp>(); in fold() local
1310 tensorflow::NodeDef* shape_op = tensorflow_graph->add_node(); in ConvertTensorFlowShapeOperator() local
5477 auto shape_op = rewriter.create<shape::ShapeOfOp>(op.getLoc(), input); in matchAndRewrite() local