Searched refs:tf_shape_ (Results 1 – 1 of 1) sorted by relevance
220 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape", &tf_shape_)); in XRTAllocateUninitializedOp()221 OP_REQUIRES_OK(ctx, TensorShapeToXLAShape(dtype_, tf_shape_, &xla_shape_)); in XRTAllocateUninitializedOp()254 TensorShape tf_shape_; variable